How much video memory do I need?

[ljx6666]

Hello!
How much video memory do I need to train this program? I only have an 11G video card, but I trained 1 epoch and automatically stopped. Is there insufficient video memory?
Thank you !

[chen1562]

try a smaller batch size, -b 64 works for me on one GPU