Revert "loss detach (pytorch#731)" (pytorch#758)

This reverts commit 391be73.
mpc001 · Apr 23, 2020 · 69d2798 · 69d2798
1 parent d431037
commit 69d2798
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.detach().item(), images.size(0))
+        losses.update(loss.item(), images.size(0))
         top1.update(acc1[0], images.size(0))
         top5.update(acc5[0], images.size(0))