Skip to content

Commit d0c5aa2

Browse files
committed
Bug fix: Error checking for DeepONet
1 parent 0a3babe commit d0c5aa2

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

deepxde/nn/deeponet_strategy.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,6 @@ class SingleOutputStrategy(DeepONetStrategy):
2727
"""Single output build strategy is the standard build method."""
2828

2929
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-
)
3430
branch = self.net.build_branch_net(layer_sizes_branch)
3531
trunk = self.net.build_trunk_net(layer_sizes_trunk)
3632
return branch, trunk

0 commit comments

Comments
 (0)