search

JDAI-CV / fast-reid

SOTA Re-identification Methods and Toolbox
Apache License 2.0
3.42k stars 837 forks source link

How to evaluation on CPU? #636

Closed mamimaka closed 2 years ago

mamimaka commented 2 years ago

我按照文档输入了指令： python3 tools/train_net.py --config-file ./configs/Market1501/bagtricks_R50.yml --eval-only \ MODEL.WEIGHTS /home/xhr/桌面/fast-reid-master/pretrained_model/2022-01-27-16-56-46-fast-reid-msmt_agw_R101-ibn.pth MODEL.DEVICE "cpu"

上面预训练模型是我自己是的路径，MODEL.DEVICE我改为了CPU

但在evaluation过程中出现了错误：

Screenshot_20220128_104208

请问该如何解决这个问题呢？

L1aoXingyu commented 2 years ago

抱歉现在不支持在 cpu 上推理和测试