wassname/TTS
1
0 Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-10-05 12:30:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

bug fix vocoder training

Browse Source
This commit is contained in:
erogol committed 2020-06-05 13:23:11 +02:00
1 parent fd8f1ecb7d
commit 189227c741
2 files changed
+6 -3
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

vocoder/train.py
+4 -1
View File
@@ -173,7 +173,10 @@ def train(model_G, criterion_G, optimizer_G, model_D, criterion_D, optimizer_D,
loss_dict = dict()
for key, value in loss_G_dict.items():
loss_dict[key] = value.item()
if isinstance(value, int):
loss_dict[key] = value
else:
loss_dict[key] = value.item()
##############################
# DISCRIMINATOR