We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0a3babe commit d0c5aa2Copy full SHA for d0c5aa2
deepxde/nn/deeponet_strategy.py
@@ -27,10 +27,6 @@ class SingleOutputStrategy(DeepONetStrategy):
27
"""Single output build strategy is the standard build method."""
28
29
def build(self, layer_sizes_branch, layer_sizes_trunk):
30
- if layer_sizes_branch[-1] != layer_sizes_trunk[-1]:
31
- raise AssertionError(
32
- "Output sizes of branch net and trunk net do not match."
33
- )
34
branch = self.net.build_branch_net(layer_sizes_branch)
35
trunk = self.net.build_trunk_net(layer_sizes_trunk)
36
return branch, trunk
0 commit comments