wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix device allocation

Browse Source
This commit is contained in:
Eren Gölge
2022-07-12 19:05:25 +02:00
committed by GitHub
parent 49bac724c0
commit bc1f93c299
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/layers/losses.py
+2 -2
View File
@@ -133,11 +133,11 @@ class SSIMLoss(torch.nn.Module):
if ssim_loss.item() > 1.0:
print(f" > SSIM loss is out-of-range {ssim_loss.item()}, setting it 1.0")
ssim_loss = torch.tensor([1.0])
ssim_loss = torch.tensor([1.0], device=ssim_loss.device)
if ssim_loss.item() < 0.0:
print(f" > SSIM loss is out-of-range {ssim_loss.item()}, setting it 0.0")
ssim_loss = torch.tensor([0.0])
ssim_loss = torch.tensor([0.0], device=ssim_loss.device)
return ssim_loss