How to train with ViT - Githubissues

JDAI-CV / fast-reid

SOTA Re-identification Methods and Toolbox

Apache License 2.0

3.42k stars 837 forks source link

How to train with ViT #653

Closed genhao3 closed 2 years ago

genhao3 commented 2 years ago

The evaluation metrics are very low when I use the original configuration of bagtricks_vit.yml for training. What am I doing wrong? Do I need to modify the code elsewhere?

peterzpy commented 2 years ago

Maybe another version of pytorch could solve this problem. I meet the same problem under pytorch1.10 and solve it by using pytoch1.8.1.

genhao3 commented 2 years ago

Maybe another version of pytorch could solve this problem. I meet the same problem under pytorch1.10 and solve it by using pytoch1.8.1.

That's great!