Update loss.py

2025-08-26 18:24:18 +00:00 · 2024-10-22 21:33:14 +08:00 · 2024-10-22 21:33:14 +08:00 · 15f2f7190f
commit 15f2f7190f
parent 442a745d8e
1 changed files with 1 additions and 1 deletions
--- a/egs/tokenizer/CODEC/encodec/loss.py
+++ b/egs/tokenizer/CODEC/encodec/loss.py
@ -309,10 +309,10 @@ class SpectrogramReconstructionLoss(torch.nn.Module):
            s = 2**i // 8
            self.wav_to_specs.append(
                Spectrogram(
-                    sample_rate=sampling_rate,
                    n_fft=s,
                    win_length=s,
                    hop_length=s // 4,
+                    window_fn=torch.hann_window,
                    normalized=True,
                    center=False,
                    pad_mode=None,