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

minor fixes

Browse Source
This commit is contained in:
zr_jin 2024-10-22 22:05:49 +08:00
parent 15f2f7190f
commit 28bd9330d7
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -143,7 +143,7 @@ class CodecDataModule:
group.add_argument(
"--input-strategy",
type=str,
default="PrecomputedFeatures",
default="AudioSamples",
help="AudioSamples or PrecomputedFeatures",
)

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

@ -300,13 +300,12 @@ class SpectrogramReconstructionLoss(torch.nn.Module):
def __init__(
self,
sampling_rate: int = 22050,
return_spec: bool = False,
):
super().__init__()
self.wav_to_specs = []
for i in range(5, 10):
s = 2**i // 8
for i in range(5, 12):
s = 2**i
self.wav_to_specs.append(
Spectrogram(
n_fft=s,