search

Jeff-Zilence / TransGeo2022

Official repository for TransGeo: Transformer Is All You Need for Cross-view Image Geo-localization
MIT License
116 stars 22 forks source link

training on single GPU #41

Closed nsa05605 closed 3 months ago

nsa05605 commented 6 months ago

Hello,

Using the provided code, can I utilize the VIGOR dataset with a single GPU?

It seems that the args.distributed is applied when either multiprocessing_distributed or world_size > 1, otherwise the training doesn't proceed.

Is it possible to train with a single GPU?

Thank you.

Jeff-Zilence commented 6 months ago

You can follow the command in the given bash script. It uses 1 GPU by default.