mchong6 / JoJoGAN

Official PyTorch repo for JoJoGAN: One Shot Face Stylization

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CUDA out of memory

songyiren98 opened this issue · comments

hello, I try to run "Train with your own style images" but got "CUDA out of memory" (2080Ti, 12G GPU memory), can you tell me how much GPU memory the Train process cost?

Training with your own images should take less than 12gb unless you're using more than one image.