Why re-ranking is running on CPU?

tinyvision / SOLIDER-REID

MIT License

55 stars 12 forks source link

Open Hocason opened 1 year ago

Hocason commented 1 year ago

The test part takes too long when re-ranking

cwhgn commented 1 year ago

We follow the code from TransReID. Thanks for your advice. It is a good suggestion to put the rerank on GPU.