Out of GPU memory with 256x256 size images

wchen342 / SketchyGAN

Code for paper "SketchyGAN: Towards Diverse and Realistic Sketch to Image Synthesis"

MIT License

131 stars 32 forks source link

Out of GPU memory with 256x256 size images #31

Open codaibk opened 5 years ago

codaibk commented 5 years ago

Dear, Dr.wchen342 It is fine with image size 64x64 pixels but the GPU memory is not enough when i tried to run it with 256x256 size images. My GPU is GTX 1080 with 8GB. What is your GPU's specification you used to run with 256x256 size images? Thanks,

wchen342 commented 5 years ago

On 1080 I think you can only run a batch size of 2 when using 256x256.