loss detach (pytorch#731)

loss detach
mpc001 · Apr 11, 2020 · 391be73 · 391be73
1 parent 80fcb66
commit 391be73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))