mirror of https://github.com/coqui-ai/TTS.git
linter fixes
parent
2b5cb24db7
commit
3c961370e7
|
@ -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']
|
||||||
|
|
|
@ -104,7 +104,6 @@ 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
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue