Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

passing reduced losses to loss dict

This commit is contained in:
erogol 2020-04-23 15:46:11 +02:00
parent 6e2c8c6537
commit 3673cc1e30
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
train.py
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':