update v2 code

WillSuen · Jun 2, 2020 · 70f5d4f · 70f5d4f
1 parent 64598e9
commit 70f5d4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/resnet_generator_v2.py b/model/resnet_generator_v2.py
@@ -42,7 +42,7 @@ def __init__(self, ch=64, z_dim=128, num_classes=10, output_dim=3):
         self.mask_regress = MaskRegressNetv2(num_w)
         self.init_parameter()
 
-    def forward(self, z, bbox, z_im=None, y=None, include_mask_loss=False):
+    def forward(self, z, bbox, z_im=None, y=None):
         b, o = z.size(0), z.size(1)
         label_embedding = self.label_embedding(y)