Merge pull request svc-develop-team#168 from Stardust-SJF/Stardust-SJ…

…F-patch-1 Fix error: math domain error when loss is too low
sundevil0405 · Apr 18, 2023 · f4ad14d · f4ad14d
2 parents 5570594 + ccbdf9e
commit f4ad14d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/train.py b/train.py
@@ -225,8 +225,7 @@ def train_and_evaluate(rank, epoch, hps, nets, optims, schedulers, scaler, loade
                 losses = [loss_disc, loss_gen, loss_fm, loss_mel, loss_kl]
                 reference_loss=0
                 for i in losses:
-                    reference_loss += math.log(i, 10)
-                reference_loss*=10
+                    reference_loss += i
                 logger.info('Train Epoch: {} [{:.0f}%]'.format(
                     epoch,
                     100. * batch_idx / len(train_loader)))