Anyone else getting memory issues?
kilickaya opened this issue · comments
❔Question
Hello! I wonder if anyone else is getting GPU memory errors even with the small model (yolos_small) ?
Additional context
I am on a 4 GPUs node with Geforce Gtx 1080 ti with 11gb memory each. I use batch size 1 as recommended. Both distributed and non-distributed versions throw the same error.
Tiny model trains smoothly without a trouble.
If there are any tips to reduce memory usage that would be awesome as well!
Hi @kilickaya, thanks for your interest in YOLOS.
We have actually answered here:#5 (comment), --use_checkpoint may help you with that.