Windows log filename bugfix

r9y9 · Mar 10, 2018 · 5214c24 · 5214c24
1 parent 747f2e0
commit 5214c24
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -952,7 +952,10 @@ def restore_parts(path, model):
 
  # Setup summary writer for tensorboard
  if log_event_path is None:
- log_event_path = "log/run-test" + str(datetime.now()).replace(" ", "_")
+ if platform.system() == "Windows":
+ log_event_path = "log/run-test" + str(datetime.now()).replace(" ", "_").replace(":","_")
+ else:
+ log_event_path = "log/run-test" + str(datetime.now()).replace(" ", "_")
  print("Los event path: {}".format(log_event_path))
  writer = SummaryWriter(log_dir=log_event_path)