Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update disabled structure

This commit is contained in:
Ayush Chaurasia 2021-07-07 17:04:52 +05:30 committed by Eren Gölge
parent f606741dc4
commit f4434da5a3
2 changed files with 21 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
TTS/trainer.py
View File

@ -174,10 +174,16 @@ class Trainer:
self.tb_logger = tb_logger self.tb_logger = tb_logger
if wandb_logger is None: if wandb_logger is None:
self.wandb_logger = WandbLogger( wandb_project_name = config.model
project=os.path(output_path).stem, if config.wandb_project_name:
name=config.model, wandb_project_name = config.wandb_project_name
wandb_logger = WandbLogger(
disabled=config.wandb_disabled,
project=wandb_project_name,
name=config.run_name,
config=config, config=config,
entity=config.wandb_entity,
) )
else: else:
self.wandb_logger = wandb_logger self.wandb_logger = wandb_logger
@ -1158,14 +1164,17 @@ def process_args(args, config=None):
# write model desc to tensorboard # write model desc to tensorboard
tb_logger.tb_add_text("model-config", f"<pre>{config.to_json()}</pre>", 0) tb_logger.tb_add_text("model-config", f"<pre>{config.to_json()}</pre>", 0)
if not config.wandb_disabled: wandb_project_name = config.model
wandb_project_name = config.model if config.wandb_project_name:
if config.wandb_project_name: wandb_project_name = config.wandb_project_name
wandb_project_name = config.wandb_project_name
wandb_logger = WandbLogger( wandb_logger = WandbLogger(
project=wandb_project_name, name=config.run_name, config=config, entity=config.wandb_entity disabled=config.wandb_disabled,
) project=wandb_project_name,
name=config.run_name,
config=config,
entity=config.wandb_entity,
)
c_logger = ConsoleLogger() c_logger = ConsoleLogger()
return config, experiment_path, audio_path, c_logger, tb_logger, wandb_logger return config, experiment_path, audio_path, c_logger, tb_logger, wandb_logger

4
TTS/utils/logging/wandb_logger.py
View File

@ -8,9 +8,9 @@ except ImportError:
class WandbLogger: class WandbLogger:
def __init__(self, **kwargs): def __init__(self, disabled=False, **kwargs):
self.run = None self.run = None
if wandb: if wandb and not disabled:
self.run = wandb.init(**kwargs) if not wandb.run else wandb.run self.run = wandb.init(**kwargs) if not wandb.run else wandb.run
self.log_dict = {} self.log_dict = {}