From 513501b32491b982e2332bd247cf2f3f4d4f11ef Mon Sep 17 00:00:00 2001 From: Grace Omotoso Date: Wed, 7 Apr 2021 18:29:48 -0500 Subject: [PATCH] tentatively made flush_secs 60 --- vissl/trainer/train_task.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vissl/trainer/train_task.py b/vissl/trainer/train_task.py index 8484d3ffb..145bd54de 100644 --- 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