Fixed RuntimeError: CUDA out of memory

lucidrains · Jan 9, 2021 · 0986c3a · 0986c3a
1 parent 147527c
commit 0986c3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lightweight_gan/lightweight_gan.py b/lightweight_gan/lightweight_gan.py
@@ -1065,7 +1065,7 @@ def train(self):
         self.steps += 1
 
     @torch.no_grad()
-    def evaluate(self, num = 0, num_image_tiles = 8, trunc = 1.0):
+    def evaluate(self, num = 0, num_image_tiles = 4, trunc = 1.0):
         self.GAN.eval()
 
         ext = self.image_extension