Skip to content

Commit 7febdb9

Browse files
authored
Fix missing argument
After the `residual_block` function was added an extra argument, the call to it on line 145 wasn't updated.
1 parent 21f4bce commit 7febdb9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

srgan/srgan.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def deconv2d(layer_input):
142142
c1 = Activation('relu')(c1)
143143

144144
# Propogate through residual blocks
145-
r = residual_block(c1)
145+
r = residual_block(c1, self.gf)
146146
for _ in range(self.n_residual_blocks - 1):
147147
r = residual_block(r, self.gf)
148148

0 commit comments

Comments
 (0)