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

updated encodec.py

Browse Source
This commit is contained in:
JinZr 2024-10-22 22:15:29 +08:00
parent 28bd9330d7
commit fe27d2ca36
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/tokenizer/CODEC/encodec/encodec.py
View File

@ -80,9 +80,7 @@ class Encodec(nn.Module):
) )
self.feature_match_loss = FeatureLoss() self.feature_match_loss = FeatureLoss()
self.wav_reconstruction_loss = WavReconstructionLoss() self.wav_reconstruction_loss = WavReconstructionLoss()
self.spec_reconstruction_loss = SpectrogramReconstructionLoss( self.spec_reconstruction_loss = SpectrogramReconstructionLoss()
sampling_rate=self.sampling_rate
)
def _forward_generator( def _forward_generator(
self, self,