about out of memory - Githubissues

heykeetae / Self-Attention-GAN

Pytorch implementation of Self-Attention Generative Adversarial Networks (SAGAN)

2.53k stars 475 forks source link

about out of memory #8

Closed shz1314 closed 6 years ago

shz1314 commented 6 years ago

Hi My GPU is 11 GB, but when i training. Always : RuntimeError: cuda runtime error (2) : out of memory at /pytorch/torch/lib/THC/generic/THCStorage.cu:58 results in the code : result = self.forward(*input, **kwargs)

cbokpark commented 6 years ago

Have you tried to clone the latest version from this repository? The latest version solved memory problem.