From 0abb5117171b305f9150feba5d9bb1c1796088b9 Mon Sep 17 00:00:00 2001
From: whoami <48873278+lsrami@users.noreply.github.com>
Date: Thu, 30 May 2024 13:34:34 +0800
Subject: [PATCH] [fix] Update train.py (#217)

Use float type when calling mel_spectrogram torch
---
 wetts/vits/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wetts/vits/train.py b/wetts/vits/train.py
index b0377f0..6f5540b 100644
--- a/wetts/vits/train.py
+++ b/wetts/vits/train.py
@@ -418,7 +418,7 @@ def train_and_evaluate(rank, local_rank, epoch, hps, nets, optims, schedulers, s
             y_mel = commons.slice_segments(
                 mel, ids_slice, hps.train.segment_size // hps.data.hop_length)
             y_hat_mel = mel_spectrogram_torch(
-                y_hat.squeeze(1),
+                y_hat.squeeze(1).float(),
                 hps.data.filter_length,
                 hps.data.n_mel_channels,
                 hps.data.sampling_rate,