From b343337ef30afac0c10d4bac46bb2cdc087e1b36 Mon Sep 17 00:00:00 2001
From: FriedRonaldo <bkjbkj12@yonsei.ac.kr>
Date: Thu, 18 Jun 2020 13:16:29 +0900
Subject: [PATCH] Update generator.py

---
 models/generator.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/models/generator.py b/models/generator.py
index 23674f7..1eb0ffe 100644
--- 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)