bytedance / R2Former

Official repository for R2Former: Unified Retrieval and Reranking Transformer for Place Recognition
Apache License 2.0
83 stars 6 forks source link

Question about setting of pitts30k training #9

Closed Graysonggg closed 1 year ago

Graysonggg commented 1 year ago

Great work! I would like to inquire about the specific parameter settings for training on the pitts30k dataset. Can you provide files similar to train_ Deit.sh and train_reranker.sh when training models only on pitts30k? Thank you!

Graysonggg commented 1 year ago

I notice that you trained the model on pitts30k mentioned in supplementary Material's Table 1. The model size is large and pitts30k is small. Can you just give more training details? Thank you.

Jeff-Zilence commented 1 year ago

Pitts30K is small and there is no need to use my global mining strategy. You can change the --mining to full and run the same command in train_reranker.sh. The deit-s model is actually very small and you should be able to get similar results with standard setting.

szhu-bytedance commented 1 year ago

I will close the issue if there is no further question.