tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update losses.py

Browse Source
This commit is contained in:
Sihan Wang
2023-12-20 00:07:15 +08:00
parent 50ad3389e8
commit 9cd9dfac6e
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
losses.py
View File

@@ -66,6 +66,7 @@ class WavLMLoss(torch.nn.Module):
self.wavlm = AutoModel.from_pretrained(model) self.wavlm = AutoModel.from_pretrained(model)
self.wd = wd self.wd = wd
self.resample = torchaudio.transforms.Resample(model_sr, slm_sr) self.resample = torchaudio.transforms.Resample(model_sr, slm_sr)
self.wavlm.eval()
def forward(self, wav, y_rec): def forward(self, wav, y_rec):
with torch.no_grad(): with torch.no_grad():