how to use gpu to run the basline/train.py

hehefan / Unsupervised-Person-Re-identification-Clustering-and-Fine-tuning

219 stars 87 forks source link

Open mod1998 opened 6 years ago

mod1998 commented 6 years ago

I succeed run the baseline/train.py,but it was runned by cpu,can I use GPU to run it??