Add missing ngf parameter

vanhuyz · Jul 11, 2019 · 79e8ec3 · 79e8ec3
1 parent 45d0713
commit 79e8ec3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -51,7 +51,7 @@ def __init__(self,
     self.G = Generator('G', self.is_training, ngf=ngf, norm=norm, image_size=image_size)
     self.D_Y = Discriminator('D_Y',
         self.is_training, norm=norm, use_sigmoid=use_sigmoid)
-    self.F = Generator('F', self.is_training, norm=norm, image_size=image_size)
+    self.F = Generator('F', self.is_training, ngf=ngf, norm=norm, image_size=image_size)
     self.D_X = Discriminator('D_X',
         self.is_training, norm=norm, use_sigmoid=use_sigmoid)