From df16bf1b18e03213efe9b22a0155b833a06c6a9c Mon Sep 17 00:00:00 2001 From: Jean-Philippe Mercier Date: Thu, 2 Nov 2017 18:02:02 -0400 Subject: gpu memory leaks A tensor is now fed to ImagePool() instead of a Variable --- util/image_pool.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'util') diff --git a/util/image_pool.py b/util/image_pool.py index 152ef5b..9f34a09 100644 --- a/util/image_pool.py +++ b/util/image_pool.py @@ -13,7 +13,7 @@ class ImagePool(): if self.pool_size == 0: return images return_images = [] - for image in images.data: + for image in images: image = torch.unsqueeze(image, 0) if self.num_imgs < self.pool_size: self.num_imgs = self.num_imgs + 1 -- cgit v1.2.3-70-g09d2