Merge pull request #108 from rkfg/patch-1

Add missing ngf parameter
vanhuyz · Jul 11, 2019 · 5279c03 · 5279c03
2 parents 45d0713 + 79e8ec3
commit 5279c03
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)