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