From b777cd26a9f89ec28daa44b6cc905b169327afca Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 2 May 2018 04:58:48 -0700 Subject: [PATCH] bug fix --- train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/train.py b/train.py index 319c8a63..b529b92d 100644 --- a/train.py +++ b/train.py @@ -103,8 +103,7 @@ def train(model, criterion, criterion_st, data_loader, optimizer, epoch): stop_target = stop_target.cuda() stop_target = stop_target.view(c.batch_size, stop_target.size(1) // c.r, -1) - import ipdb; ipdb.set_trace() - stop_target = (stop_target.sum(1) > 0.0).long() + stop_target = (stop_target.sum(2) > 0.0).long() # create attention mask if c.mk > 0.0: