search

mchong6 / JoJoGAN

Official PyTorch repo for JoJoGAN: One Shot Face Stylization
MIT License
1.42k stars 206 forks source link

CUDA out of memory #25

Closed songyiren98 closed 2 years ago

songyiren98 commented 2 years ago

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?

mchong6 commented 2 years ago

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