VRAM

[YaangHailong]

I'm having an OOM error with a 12GB GPU and batch size 8. For those who ran successfully, could you share your GPU model and VRAM so we can compare?

[Ayews]

Our training was completed on an RTX 3090 GPU. You can gradually reduce the batch size until there is no OOM, and the difference in batch size will not have a significant impact on performance.