diff --git a/fltk/core/distributed/client.py b/fltk/core/distributed/client.py index cc93b928..3001db4a 100644 --- a/fltk/core/distributed/client.py +++ b/fltk/core/distributed/client.py @@ -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): """