From 9e30f4aae86a2bb991e12aebb49ba71ef5390b6e Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Tue, 14 May 2019 22:35:11 +0200 Subject: [PATCH] bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index b3eb4f39..28b8b3f0 100644 --- a/train.py +++ b/train.py @@ -178,7 +178,7 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler, postnet_loss = reduce_tensor(postnet_loss.data, num_gpus) decoder_loss = reduce_tensor(decoder_loss.data, num_gpus) loss = reduce_tensor(loss.data, num_gpus) - stop_loss = reduce_tensor(stop_loss.data, num_gpus) + stop_loss = reduce_tensor(stop_loss.data, num_gpus) if c.stopnet else stop_loss if args.rank == 0: avg_postnet_loss += float(postnet_loss.item())