CUDA out of memory
songyiren98 opened this issue · comments
songyiren98 commented
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?
Min Jin Chong commented
Training with your own images should take less than 12gb unless you're using more than one image.