zkcys001 / UDAStrongBaseline

Open-source stronger baseline for unsupervised or domain adaptive object re-ID.
142 stars 21 forks source link

GLT_kmeans_duke2market raise StopIteration error #7

Closed woocoder closed 3 years ago

woocoder commented 3 years ago

when I follow the steps to run the code. The stage 1 is ok, but the stage 2 always raise StopIteration error. image image And I check the variables in GLT_kmeans_duke2market.py, I found the error begin from here: image The values of label is same, And I don't know why. image

woocoder commented 3 years ago

emmm, the reason looks like my gpu is 3090 and cuda 11. So the faiss-gpu in pip doesn't work. when I install faiss follow the issue: https://github.com/facebookresearch/faiss/issues/1629#issuecomment-780509841. The error never raise.

image