![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
torch
文章平均质量分 79
kongkongqixi
技术技术技术,非要科班嘛
展开
-
copying a param of torch.Size([128]) from checkpoint, where the shape is torch.Size([1, 128])解决方式
关于torch的在加载bn_inception、x_inception时出现copying a param of torch.Size([128]) from checkpoint, where the shape is torch.Size([1, 128])类型的错误,观察发现可以用过修改加载模型中权重维度解决问题。例如:bn_inception中都是两维元组的数据转为一维:如(1,128...原创 2019-03-05 15:11:23 · 9269 阅读 · 1 评论 -
pytorch关于多块gpu使用总结,报错AssertionError: Invalid device id
pytorch默认使用gpu编号为device:0的设备,可以使用 torch.nn.DataParallel(model, device_ids=[0, 1]) 对模型进行制定gpu指定编号多gpu训练,必须要有编号为device:0的gpu,不然会报AssertionError: Invalid device id错误; 当gpu编号为device:0的设备被占用时,指定其他编号gpu...原创 2019-09-03 16:45:55 · 24395 阅读 · 14 评论