Commit 76c6a42a authored by Andrei Roibu's avatar Andrei Roibu
Browse files

commented out the removal of the logs dir if path exists when using last checkpoint

parent 55ffe4c0
...@@ -55,11 +55,11 @@ class LogWriter(): ...@@ -55,11 +55,11 @@ class LogWriter():
logs_directory, experiment_name, "validation") logs_directory, experiment_name, "validation")
# If the logs directory exist, we clear their contents to allow new logs to be created # If the logs directory exist, we clear their contents to allow new logs to be created
if not use_last_checkpoint: # if not use_last_checkpoint:
if os.path.exists(training_logs_directory): # if os.path.exists(training_logs_directory):
shutil.rmtree(training_logs_directory) # shutil.rmtree(training_logs_directory)
if os.path.exists(validation_logs_directory): # if os.path.exists(validation_logs_directory):
shutil.rmtree(validation_logs_directory) # shutil.rmtree(validation_logs_directory)
self.log_writer = { self.log_writer = {
'train': SummaryWriter(logdir=training_logs_directory), 'train': SummaryWriter(logdir=training_logs_directory),
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment