- 博客(2)
- 收藏
- 关注
RSS订阅原创 Pytorch训练问题：AssertionError: Invalid device id
在Linux中使用显卡训练网络时，一般会通过device id来确定使用的显卡。我们从GitHub上获取的源码中的device id和我们本地的device id肯定不一致，所以训练时一定要注意device id修改。以下示例：源码：model = nn.DataParallel( model.cuda(), device_ids=[0,1] 源码中使用了id为0和1 的显卡进行训练。本地训练报错： AssertionError: In...
2021-08-02 09:05:41
6755
2
原创 python--归一化图像二值化
备注：在python中使用opencv好像无法打开归一化图片，故推荐PIL中Image方法。import osfrom PIL import Imageimg_path = 'C:/Users/**/Desktop/test_result/artery_test_results'save_path = 'C:/Users/**/Desktop/test_result/artery_test_process'img_list = os.listdir(img_path)for im.
2021-07-20 19:38:50
355
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人