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

passing reduced losses to loss dict

Browse Source
This commit is contained in:
erogol committed 2020-04-23 15:46:11 +02:00
1 parent 6e2c8c6537
commit 3673cc1e30
1 file changed
+3 -3
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+3 -3
View File
@@ -372,10 +372,10 @@ def evaluate(model, criterion, ap, global_step, epoch):
# aggregate losses from processes
if num_gpus > 1:
postnet_loss = reduce_tensor(loss_dict['postnet_loss'].data, num_gpus)
decoder_loss = reduce_tensor(loss_dict['decoder_loss'].data, num_gpus)
loss_dict['postnet_loss'] = reduce_tensor(loss_dict['postnet_loss'].data, num_gpus)
loss_dict['decoder_loss'] = reduce_tensor(loss_dict['decoder_loss'].data, num_gpus)
if c.stopnet:
stopnet_loss = reduce_tensor(loss_dict['stopnet_loss'].data, num_gpus)
loss_dict['stopnet_loss'] = reduce_tensor(loss_dict['stopnet_loss'].data, num_gpus)
keep_avg.update_values({
'avg_postnet_loss':