removed .item() from the averaging

IAHispano · Jan 4, 2025 · de7d1eb · de7d1eb
1 parent b59f4af
commit de7d1eb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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(