[fix] Update train.py (#217)

Use float type when calling mel_spectrogram torch
wenet-e2e · May 30, 2024 · 0abb511 · 0abb511
1 parent 97b83e8
commit 0abb511
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,