mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-12 11:32:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update vits.py

Browse Source 
FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', args...)` instead.
This commit is contained in:
Masoud Azizi 2024-11-30 18:55:13 +03:30 committed by GitHub
parent b7b65205d1
commit 046d257570
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/ljspeech/TTS/vits/vits.py
View File

@ -25,7 +25,7 @@ from loss import (
KLDivergenceLoss, KLDivergenceLoss,
MelSpectrogramLoss, MelSpectrogramLoss,
) )
from torch.cuda.amp import autocast from torch.amp import autocast
from utils import get_segments from utils import get_segments
AVAILABLE_GENERATERS = { AVAILABLE_GENERATERS = {
@ -410,7 +410,7 @@ class VITS(nn.Module):
p = self.discriminator(speech_) p = self.discriminator(speech_)
# calculate losses # calculate losses
with autocast(enabled=False): with autocast('cuda',enabled=False):
if not return_sample: if not return_sample:
mel_loss = self.mel_loss(speech_hat_, speech_) mel_loss = self.mel_loss(speech_hat_, speech_)
else: else:
@ -518,7 +518,7 @@ class VITS(nn.Module):
p = self.discriminator(speech_) p = self.discriminator(speech_)
# calculate losses # calculate losses
with autocast(enabled=False): with autocast('cuda',enabled=False):
real_loss, fake_loss = self.discriminator_adv_loss(p_hat, p) real_loss, fake_loss = self.discriminator_adv_loss(p_hat, p)
loss = real_loss + fake_loss loss = real_loss + fake_loss