Update train.py

mazzzystar · Mar 12, 2019 · 3c99f1d · 3c99f1d
1 parent 910a244
commit 3c99f1d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/train.py b/train.py
@@ -171,10 +171,10 @@
                 D_wass_valid = D_wass_valid.cpu()
 
             # Record costs
-            D_cost_train_epoch.append(D_cost_train.data.numpy()
-            D_wass_train_epoch.append(D_wass_train.data.numpy()
-            D_cost_valid_epoch.append(D_cost_valid.data.numpy()
-            D_wass_valid_epoch.append(D_wass_valid.data.numpy()
+            D_cost_train_epoch.append(D_cost_train.data.numpy())
+            D_wass_train_epoch.append(D_wass_train.data.numpy())
+            D_cost_valid_epoch.append(D_cost_valid.data.numpy())
+            D_wass_valid_epoch.append(D_wass_valid.data.numpy())
 
         #############################
         # (3) Train Generator