diff --git a/TTS/config/shared_configs.py b/TTS/config/shared_configs.py index ebc18e8c..4a62735d 100644 --- a/TTS/config/shared_configs.py +++ b/TTS/config/shared_configs.py @@ -236,7 +236,7 @@ class BaseTrainingConfig(Coqpit): Defaults to ```False```. project_name (str): - Name of the W&B project. Defaults to config.model + Name of the project. Defaults to config.model wandb_entity (str): Name of W&B entity/team. Enables collaboration across a team or org. diff --git a/TTS/trainer.py b/TTS/trainer.py index f0a2b18e..6acbe051 100644 --- a/TTS/trainer.py +++ b/TTS/trainer.py @@ -168,7 +168,6 @@ class Trainer: if self.dashboard_logger is None: if config.dashboard_logger == "tensorboard": self.dashboard_logger = TensorboardLogger(output_path, model_name=config.model) - self.dashboard_logger.add_text("model-config", f"
{config.to_json()}
", 0) elif config.dashboard_logger == "wandb": project_name = config.model @@ -181,6 +180,8 @@ class Trainer: config=config, entity=config.wandb_entity, ) + + self.dashboard_logger.add_text("model-config", f"
{config.to_json()}
", 0) if not self.config.log_model_step: self.config.log_model_step = self.config.save_step @@ -1147,7 +1148,6 @@ def process_args(args, config=None): if config.dashboard_logger == "tensorboard": dashboard_logger = TensorboardLogger(config.output_path, model_name=config.model) - dashboard_logger.add_text("model-config", f"
{config.to_json()}
", 0) elif config.dashboard_logger == "wandb": project_name = config.model @@ -1160,7 +1160,7 @@ def process_args(args, config=None): config=config, entity=config.wandb_entity, ) - + dashboard_logger.add_text("model-config", f"
{config.to_json()}
", 0) c_logger = ConsoleLogger() return config, experiment_path, audio_path, c_logger, dashboard_logger diff --git a/TTS/utils/logging/wandb_logger.py b/TTS/utils/logging/wandb_logger.py index f2fb6e1d..5fcab00f 100644 --- a/TTS/utils/logging/wandb_logger.py +++ b/TTS/utils/logging/wandb_logger.py @@ -86,7 +86,7 @@ class WandbLogger: self.dict_to_figure(f"{self.model_name}_TestFigures", figures) def add_text(self, title, text, step): - self.log_dict[title] = wandb.HTML(f"

{text}

") + pass def flush(self): if self.run: