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( group.add_argument(
"--input-strategy", "--input-strategy",
type=str, type=str,
default="PrecomputedFeatures", default="AudioSamples",
help="AudioSamples or PrecomputedFeatures", help="AudioSamples or PrecomputedFeatures",
) )

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

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