Fix missing argument

After the `residual_block` function was added an extra argument, the call to it on line 145 wasn't updated.
zhxu-astro · Aug 20, 2018 · 7febdb9 · 7febdb9
1 parent 21f4bce
commit 7febdb9
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)