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

linter fixes

pull/367/head
Eren Gölge 2021-02-09 11:43:17 +00:00 committed by Eren Gölge
parent 2b5cb24db7
commit 3c961370e7
3 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/bin/train_speedy_speech.py
View File

@ -515,7 +515,7 @@ def main(args): # pylint: disable=redefined-outer-name
train_avg_loss_dict, global_step = train(train_loader, model, criterion, optimizer, train_avg_loss_dict, global_step = train(train_loader, model, criterion, optimizer,
scheduler, ap, global_step, scheduler, ap, global_step,
epoch) epoch)
eval_avg_loss_dict = evaluate(eval_loader , model, criterion, ap, eval_avg_loss_dict = evaluate(eval_loader, model, criterion, ap,
global_step, epoch) global_step, epoch)
c_logger.print_epoch_end(epoch, eval_avg_loss_dict) c_logger.print_epoch_end(epoch, eval_avg_loss_dict)
target_loss = train_avg_loss_dict['avg_loss'] target_loss = train_avg_loss_dict['avg_loss']

5
TTS/utils/arguments.py
View File

@ -104,8 +104,7 @@ def get_last_checkpoint(path):
pass pass
if last_checkpoint is None: if last_checkpoint is None:
raise ValueError(f"No checkpoints in {path}!") raise ValueError(f"No checkpoints in {path}!")
else: return last_checkpoint
return last_checkpoint
def process_args(args, model_type): def process_args(args, model_type):
@ -193,7 +192,7 @@ def process_args(args, model_type):
if args.restore_path: if args.restore_path:
new_fields["restore_path"] = args.restore_path new_fields["restore_path"] = args.restore_path
new_fields["github_branch"] = get_git_branch() new_fields["github_branch"] = get_git_branch()
copy_model_files(c, args.config_path, copy_model_files(c, args.config_path,
out_path, new_fields) out_path, new_fields)
os.chmod(audio_path, 0o775) os.chmod(audio_path, 0o775)
os.chmod(out_path, 0o775) os.chmod(out_path, 0o775)