From 39d1adce1fda14b4c5d5def906e3584555157099 Mon Sep 17 00:00:00 2001 From: Yonghye Kwon Date: Wed, 11 Mar 2020 11:52:49 +0900 Subject: [PATCH] loss detach loss detach --- imagenet/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imagenet/main.py b/imagenet/main.py index 2a7fc13f6c..c6e338f0d7 100644 --- a/imagenet/main.py +++ b/imagenet/main.py @@ -289,7 +289,7 @@ def train(train_loader, model, criterion, optimizer, epoch, args): # measure accuracy and record loss acc1, acc5 = accuracy(output, target, topk=(1, 5)) - losses.update(loss.item(), images.size(0)) + losses.update(loss.detach().item(), images.size(0)) top1.update(acc1[0], images.size(0)) top5.update(acc5[0], images.size(0))