mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update loss.py

Browse Source
This commit is contained in:
zr_jin 2024-10-22 21:33:14 +08:00
parent 442a745d8e
commit 15f2f7190f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/tokenizer/CODEC/encodec/loss.py
View File

@ -309,10 +309,10 @@ class SpectrogramReconstructionLoss(torch.nn.Module):
s = 2**i // 8 s = 2**i // 8
self.wav_to_specs.append( self.wav_to_specs.append(
Spectrogram( Spectrogram(
sample_rate=sampling_rate,
n_fft=s, n_fft=s,
win_length=s, win_length=s,
hop_length=s // 4, hop_length=s // 4,
window_fn=torch.hann_window,
normalized=True, normalized=True,
center=False, center=False,
pad_mode=None, pad_mode=None,