Skip to content

Commit

Permalink
Update client so only ROOT gets logging object
Browse files Browse the repository at this point in the history
  • Loading branch information
JMGaljaard committed Sep 6, 2022
1 parent d43d1fc commit 945339e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions fltk/core/distributed/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,9 +80,9 @@ def prepare_learner(self, distributed: bool = False) -> None:
self.learning_params.scheduler_gamma,
self.learning_params.min_lr)

if self.config.execution_config.tensorboard.active:
if self.config.execution_config.tensorboard.active and self._id == 0:
self.tb_writer = SummaryWriter(
str(self.config.get_log_path(self._task_id, self._id, self.learning_params.model)))
str(self.config.get_log_path(self._task_id, self._id, self.learning_params)))

def stop_learner(self):
"""
Expand Down

0 comments on commit 945339e

Please sign in to comment.