- 博客(2)
- 收藏
- 关注
原创 记录今天遇到的一个坑
记录今天遇到的一个坑 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 173 1
原创 记录跑HardNet时报的错误
记录跑HardNet时报的错误 从HardNet论文中的github链接下载的代码, 运行hardnet.py后开始下载数据集 下载之后程序会自动生成.pt文件(我目前还不清楚为什么要生成.pt文件) 就在生成yosmite.pt后报了个错 百度了下没有这个错误。。。 解决:手动将yosmite文件夹中的图片拿出去一部分(大概接近一半),然后程序就可以正常运行 motivation:我比较...
2019-11-16 09:44:48 315
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人