search

layumi / Person_reID_baseline_pytorch

:bouncing_ball_person: Pytorch ReID: A tiny, friendly, strong pytorch implement of person re-id / vehicle re-id baseline. Tutorial 👉https://github.com/layumi/Person_reID_baseline_pytorch/tree/master/tutorial
https://www.zdzheng.xyz
MIT License
4.08k stars 1k forks source link

only cpu #411

Open xc597 opened 3 months ago

xc597 commented 3 months ago

When running test.py, can I only set the CPU to run? If so, how should I do it?

layumi commented 3 months ago

Hi @xc597 I think it is Okay. You need to remove all .cuda() and specify .cpu().