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

linter fix

Browse Source
This commit is contained in:
erogol
2020-06-15 19:04:05 +02:00
parent ddfede1f52
commit e8a59db839
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+2 -2
View File
@@ -198,7 +198,7 @@ def train(model, criterion, optimizer, optimizer_st, scheduler,
# detach loss values
loss_dict_new = dict()
for key, value in loss_dict.items():
if isinstance(value, int) or isinstance(value, float):
if isinstance(value, (int, float)):
loss_dict_new[key] = value
else:
loss_dict_new[key] = value.item()
@@ -336,7 +336,7 @@ def evaluate(model, criterion, ap, global_step, epoch):
# detach loss values
loss_dict_new = dict()
for key, value in loss_dict.items():
if isinstance(value, int) or isinstance(value, float):
if isinstance(value, (int, float)):
loss_dict_new[key] = value
else:
loss_dict_new[key] = value.item()