Update train_second.py

fix yl4579#95 (comment)
ishine · Nov 28, 2023 · 034404f · 034404f
1 parent 17c6b61
commit 034404f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: