updated encodec.py

This commit is contained in:
JinZr 2024-10-22 22:15:29 +08:00
parent 28bd9330d7
commit fe27d2ca36

View File

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