search

Cysu / open-reid

Open source person re-identification library in python
https://cysu.github.io/open-reid/
MIT License
1.34k stars 349 forks source link

Do this project support pytorch 1.0.1? #79

Open bolinpu opened 5 years ago

krxat commented 5 years ago

Yes, it does. Only change the 33rd line in reid/trainers.py from losses.update(loss.data[0], targets.size(0)) to losses.update(loss.item(), targets.size(0))