diff --git a/train_second.py b/train_second.py index 848de51f..1b5bffe6 100644 --- a/train_second.py +++ b/train_second.py @@ -676,7 +676,7 @@ def main(config_path): logger.info('Validation loss: %.3f, Dur loss: %.3f, F0 loss: %.3f' % (loss_test / iters_test, loss_align / iters_test, loss_f / iters_test) + '\n\n\n') print('\n\n\n') writer.add_scalar('eval/mel_loss', loss_test / iters_test, epoch + 1) - writer.add_scalar('eval/dur_loss', loss_test / iters_test, epoch + 1) + writer.add_scalar('eval/dur_loss', loss_align / iters_test, epoch + 1) writer.add_scalar('eval/F0_loss', loss_f / iters_test, epoch + 1) if epoch < joint_epoch: