amaralibey / Bag-of-Queries

BoQ: A Place is Worth a Bag of learnable Queries (CVPR 2024)
MIT License
86 stars 6 forks source link

The Hyper-parameters for training the DinoV2+BoQ? #6

Closed Frost24K closed 4 months ago

Frost24K commented 4 months ago

Hi, Great work! Can you provide the concrete information in training BoQ with DinoV2 backbone? Such as the learning rate, scheduler options, train epochs, etc. Thank you!

amaralibey commented 4 months ago

Hi @Frost24K,

The model we shared has been trained on GSV-Cities with:

The VRAM usage when employing batches of size 160x4 is ~31GB. However, I've got similar performance using batch size of 100x4, which requieres ~19GB of VRAM.

Best, Amar

Frost24K commented 4 months ago

Hi @amaralibey, Thank you for your prompt response !