open-reid icon indicating copy to clipboard operation
open-reid copied to clipboard

Published 20 hours ago • verified publisher icon Cysu

How much video memory do I need?

Open ljx6666 opened this issue 5 years ago • 1 comments

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 !

ljx6666 avatar Dec 01 '19 08:12 ljx6666

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

chen1562 avatar Jan 29 '20 18:01 chen1562