Anyone else getting memory issues?

hustvl / YOLOS

[NeurIPS 2021] You Only Look at One Sequence

https://arxiv.org/abs/2106.00666

MIT License

827 stars 118 forks source link

Anyone else getting memory issues? #6

Closed kilickaya closed 3 years ago

kilickaya commented 3 years ago

❔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!

LegendBC commented 3 years ago

Hi @kilickaya, thanks for your interest in YOLOS. We have actually answered here:https://github.com/hustvl/YOLOS/issues/5#issuecomment-867533669, --use_checkpoint may help you with that.