MoVQGAN icon indicating copy to clipboard operation
MoVQGAN copied to clipboard

cuda out of memory

Open Droliven opened this issue 1 year ago • 0 comments

I modified the 67M version configs to train the network with 384*384 imgs, the size of z changed to 48*48*4. However it shown "cuda out of memory", even on an A100*80G GPU

Droliven avatar Feb 06 '24 07:02 Droliven