Update generator.py

clovaai · Jun 18, 2020 · b343337 · b343337
1 parent 1465e24
commit b343337
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/models/generator.py b/models/generator.py
@@ -44,10 +44,10 @@ def __init__(self, img_size=128, sty_dim=64, n_res=2, use_sn=False):
 
         self.apply(weights_init('kaiming'))
 
-    def forward(self, x_cnt, x_sty): 
-        print("FORWARD of GENERATOR is not implemented")
-        exit(-6)
-        return None
+    def forward(self, x_src, s_ref):
+        c_src = self.cnt_encoder(x_src)
+        x_out = self.decode(c_src, s_ref)
+        return x_out
 
     def decode(self, cnt, sty):
         adapt_params = self.mlp(sty)