运行 parsing_reid/runs/main.py报错
daiguangzhao opened this issue · 2 comments
daiguangzhao commented
daiguangzhao commented
感谢您的关注,打扰了,第一个问题已经解决了,需要将math_tools.py 142行改成->distmat = distmat.cpu().numpy()就行了
但是,main.py 53行的cfg.data.batch_size()=
无论设置多少,我的显存都没有变化,试过单卡,和双卡。显存都是没变化。。。。。
silverbulletmdc commented
您好。第一个问题是最近在测大数据集的时候对代码做了一些改动造成的,后面会做修复。关于第二个问题,配置文件中的batch_size会替换掉代码中的batch_size