Permalink
Browse files

Fix memory leak in pytorch_imagenet_resnet50.py example (#641)

  • Loading branch information...
una-dinosauria authored and alsrgv committed Nov 22, 2018
1 parent d0281be commit 6bac3dde71a7c5335932eba725c547a3b447e8d7
Showing with 1 addition and 1 deletion.
  1. +1 −1 examples/pytorch_imagenet_resnet50.py
@@ -241,7 +241,7 @@ def __init__(self, name):
self.n = torch.tensor(0.)
def update(self, val):
self.sum += hvd.allreduce(val.cpu(), name=self.name)
self.sum += hvd.allreduce(val.detach().cpu(), name=self.name)
self.n += 1
@property

0 comments on commit 6bac3dd

Please sign in to comment.