How much video memory do I need?

Cysu / open-reid

Open source person re-identification library in python

https://cysu.github.io/open-reid/

MIT License

1.33k stars 349 forks source link

How much video memory do I need? #88

Open ljx6666 opened 4 years ago

ljx6666 commented 4 years ago

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 commented 4 years ago

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