Replaces #159

closes #159
r9y9 · Jun 11, 2019 · 1c75289 · 1c75289
1 parent 5897eed
commit 1c75289
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -968,7 +968,7 @@ def restore_parts(path, model):
         else:
             log_event_path = "log/run-test" + str(datetime.now()).replace(" ", "_")
     print("Log event path: {}".format(log_event_path))
-    writer = SummaryWriter(log_dir=log_event_path)
+    writer = SummaryWriter(log_event_path)
 
     # Train!
     try: