mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 10:16:14 +00:00
updated encodec.py
This commit is contained in:
parent
28bd9330d7
commit
fe27d2ca36
@ -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,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user