From 3673cc1e30f38137afdcb1be3160ef6dff4127d7 Mon Sep 17 00:00:00 2001 From: erogol Date: Thu, 23 Apr 2020 15:46:11 +0200 Subject: [PATCH] passing reduced losses to loss dict --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index 46e2c43f..328e81e8 100644 --- a/train.py +++ b/train.py @@ -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':