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

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,
scheduler, ap, global_step,
epoch)
eval_avg_loss_dict = evaluate(eval_loader , model, criterion, ap,
eval_avg_loss_dict = evaluate(eval_loader, model, criterion, ap,
global_step, epoch)
c_logger.print_epoch_end(epoch, eval_avg_loss_dict)
target_loss = train_avg_loss_dict['avg_loss']

View File

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