Add map_location device to support loading model on CPU-only devices

amaralibey / Bag-of-Queries

BoQ: A Place is Worth a Bag of learnable Queries (CVPR 2024)

MIT License

96 stars 6 forks source link

Add map_location device to support loading model on CPU-only devices #16

Closed OliverGrainge closed 1 week ago

OliverGrainge commented 2 weeks ago

Added map_location='cpu' to ensure the model can be loaded on devices without CUDA. This change allows compatibility with CPU-only environments, improving accessibility and broadening device support for users without GPU resources.