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: