diff --git a/train.py b/train.py index 46091d4b..30cebb7a 100644 --- 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)