diff --git a/imagenet/main.py b/imagenet/main.py
index c6e338f0d7..2a7fc13f6c 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.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))