Fix logger append.

rcamino · Aug 20, 2018 · 8cbe913 · 8cbe913
1 parent 5e6146d
commit 8cbe913
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 7 deletions.
diff --git a/multi_categorical_gans/methods/arae/trainer.py b/multi_categorical_gans/methods/arae/trainer.py
@@ -63,7 +63,7 @@ def train(autoencoder,
     optim_gen = Adam(generator.parameters(), weight_decay=l2_regularization, lr=learning_rate)
     optim_disc = Adam(discriminator.parameters(), weight_decay=l2_regularization, lr=learning_rate)
 
-    logger = Logger(output_loss_path)
+    logger = Logger(output_loss_path, append=start_epoch > 0)
 
     for epoch_index in range(start_epoch, num_epochs):
         logger.start_timer()

diff --git a/multi_categorical_gans/methods/mc_gumbel/trainer.py b/multi_categorical_gans/methods/mc_gumbel/trainer.py
@@ -46,7 +46,7 @@ def train(generator,
 
     criterion = BCELoss()
 
-    logger = Logger(output_loss_path)
+    logger = Logger(output_loss_path, append=start_epoch > 0)
 
     for epoch_index in range(start_epoch, num_epochs):
         logger.start_timer()

diff --git a/multi_categorical_gans/methods/mc_wgan_gp/trainer.py b/multi_categorical_gans/methods/mc_wgan_gp/trainer.py
@@ -44,7 +44,7 @@ def train(generator,
     optim_gen = Adam(generator.parameters(), weight_decay=l2_regularization, lr=learning_rate)
     optim_disc = Adam(discriminator.parameters(), weight_decay=l2_regularization, lr=learning_rate)
 
-    logger = Logger(output_loss_path)
+    logger = Logger(output_loss_path, append=start_epoch > 0)
 
     for epoch_index in range(start_epoch, num_epochs):
         logger.start_timer()

diff --git a/multi_categorical_gans/methods/medgan/pre_trainer.py b/multi_categorical_gans/methods/medgan/pre_trainer.py
@@ -37,7 +37,7 @@ def pre_train(autoencoder,
 
     optim = Adam(autoencoder.parameters(), weight_decay=l2_regularization, lr=learning_rate)
 
-    logger = Logger(output_loss_path)
+    logger = Logger(output_loss_path, append=start_epoch > 0)
 
     for epoch_index in range(start_epoch, num_epochs):
         logger.start_timer()

diff --git a/multi_categorical_gans/methods/medgan/trainer.py b/multi_categorical_gans/methods/medgan/trainer.py
@@ -51,7 +51,7 @@ def train(autoencoder,
 
     criterion = BCELoss()
 
-    logger = Logger(output_loss_path)
+    logger = Logger(output_loss_path, append=start_epoch > 0)
 
     for epoch_index in range(start_epoch, num_epochs):
         logger.start_timer()

diff --git a/multi_categorical_gans/utils/logger.py b/multi_categorical_gans/utils/logger.py
@@ -10,8 +10,8 @@ class Logger(object):
 
     start_time = None
 
-    def __init__(self, output_path):
-        if os.path.exists(output_path) and os.path.getsize(output_path) > 0:
+    def __init__(self, output_path, append=False):
+        if append and os.path.exists(output_path) and os.path.getsize(output_path) > 0:
             self.output_file = open(output_path, "a")
             self.output_writer = csv.DictWriter(self.output_file, fieldnames=self.CSV_COLUMNS)
         else: