Skip to content
This repository was archived by the owner on Jan 2, 2021. It is now read-only.

Commit 1ded2ae

Browse files
committed
Fix for residual branch not being connected correctly.
1 parent cb55929 commit 1ded2ae

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

enhance.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -315,7 +315,9 @@ def make_layer(self, name, input, units, filter_size=(3,3), stride=(1,1), pad=(1
315315
def make_block(self, name, input, units):
316316
self.make_layer(name+'-A', input, units, alpha=0.25)
317317
self.make_layer(name+'-B', self.last_layer(), units, alpha=1.0)
318-
return ElemwiseSumLayer([input, self.last_layer()]) if args.generator_residual else self.last_layer()
318+
if args.generator_residual:
319+
self.network[name+'-R'] = ElemwiseSumLayer([input, self.last_layer()])
320+
return self.last_layer()
319321

320322
def setup_generator(self, input, config):
321323
for k, v in config.items(): setattr(args, k, v)

0 commit comments

Comments
 (0)