akanimax / pro_gan_pytorch

Unofficial PyTorch implementation of the paper titled "Progressive growing of GANs for improved Quality, Stability, and Variation"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Memory issues/requirements

MaliRobot opened this issue · comments

Hi, I am trying to work with grid of 3*3 images in 1024 resolution (each image in grid is 1024px). Somehow it is not enough, even with GPU with 24GB video memory. I may do one image at 1024 during which memory consumption is ~ 16GB, but if I add more it jumps quickly over the limit.

I am working with small set of images (around 200).

Reading around here I have impression that something is not right, some people are making huge grids with less video memory, but not me :( Any tips?