From de7d1eb3f1d2d48dfb1e244c51cee39ade5035bb Mon Sep 17 00:00:00 2001 From: AznamirWoW <101997116+AznamirWoW@users.noreply.github.com> Date: Fri, 3 Jan 2025 19:01:25 -0500 Subject: [PATCH] removed .item() from the averaging --- rvc/train/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rvc/train/train.py b/rvc/train/train.py index a407a1f1..db75260d 100644 --- a/rvc/train/train.py +++ b/rvc/train/train.py @@ -747,8 +747,8 @@ def train_and_evaluate( # Logging and checkpointing if rank == 0: - avg_losses["disc_loss_queue"].append(epoch_disc_sum.item() / len(train_loader)) - avg_losses["gen_loss_queue"].append(epoch_gen_sum.item() / len(train_loader)) + avg_losses["disc_loss_queue"].append(epoch_disc_sum / len(train_loader)) + avg_losses["gen_loss_queue"].append(epoch_gen_sum / len(train_loader)) # used for tensorboard chart - all/mel mel = spec_to_mel_torch(