- 博客(2)
- 收藏
- 关注
RSS订阅原创 记录今天遇到的一个坑
记录今天遇到的一个坑 pytorch中的device= torch.device()函数，如果放在os.environ[‘CUDA_VISIBLE_DEVICES’] = args.gpu_id之前，无论os中设置用几号gpu，都默认为gpu 0 要把device= torch.device()函数放在os.environ[‘CUDA_VISIBLE_DEVICES’] = args.gpu_id之后用 ...
2020-10-18 20:18:31
143
1
原创 记录跑HardNet时报的错误
记录跑HardNet时报的错误 从HardNet论文中的github链接下载的代码， 运行hardnet.py后开始下载数据集 下载之后程序会自动生成.pt文件（我目前还不清楚为什么要生成.pt文件） 就在生成yosmite.pt后报了个错 百度了下没有这个错误。。。 解决：手动将yosmite文件夹中的图片拿出去一部分（大概接近一半），然后程序就可以正常运行 motivation：我比较...
2019-11-16 09:44:48
297
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人