How much GPU memory required to train?

serizba / salad

Optimal Transport Aggregation for Visual Place Recognition

GNU General Public License v3.0

155 stars 16 forks source link

Closed gmberton closed 8 months ago

gmberton commented 8 months ago

Hi, how many GB of GPU memory does SALAD need to train?

serizba commented 8 months ago

Hi @gmberton

I am using a RTX 3090 for training with 24GB of VRAM. However, with batch size 60 it usually stays under 20GB.

gmberton commented 8 months ago

Thank you!