Merge pull request eriklindernoren#77 from abiro/patch-1

Fix missing argument
zhxu-astro · Aug 31, 2018 · fcddb21 · fcddb21
2 parents 21f4bce + 7febdb9
commit fcddb21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/srgan/srgan.py b/srgan/srgan.py
@@ -142,7 +142,7 @@ def deconv2d(layer_input):
         c1 = Activation('relu')(c1)
 
         # Propogate through residual blocks
-        r = residual_block(c1)
+        r = residual_block(c1, self.gf)
         for _ in range(self.n_residual_blocks - 1):
             r = residual_block(r, self.gf)