tentatively made flush_secs 60

MLH-Fellowship · Apr 8, 2021 · 513501b · 513501b
1 parent 8ff7cae
commit 513501b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vissl/trainer/train_task.py b/vissl/trainer/train_task.py
@@ -119,12 +119,12 @@ def initiate_vissl_event_storage(self):
         create_event_storage()
         self._event_storage = get_event_storage()
 
-    def build_event_storage_writers(self, config: AttrDict):
+    def build_event_storage_writers(self):
         from vissl.utils.events import JsonWriter, TensorboardWriter
 
         self.event_storage_writers = [
             JsonWriter(f"{self.checkpoint_folder}/stdout.json"),
-            TensorboardWriter(config.log_dir, config.flush_secs),
+            TensorboardWriter("./checkpoints", 60),
         ]
 
     @property