摘要: https://www.jianshu.com/p/54996f2808e1 比较详细的FPN教程,参考 阅读全文
posted @ 2020-04-01 23:42 Wind·Chaser 阅读(2422) 评论(0) 推荐(0)
摘要: 这行报错predict = model(Variable(x_train)) RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #4 ' 阅读全文
posted @ 2020-04-01 23:38 Wind·Chaser 阅读(1511) 评论(0) 推荐(0)
摘要: 运行Faster rcnn FPN网络,出现的bug:RuntimeError: reciprocal is not implemented for type torch.cuda.LongTensor 解决办法: anchor_target_layer_fpn.py", line 136: num 阅读全文
posted @ 2020-04-01 20:09 Wind·Chaser 阅读(1037) 评论(0) 推荐(0)
摘要: 如何解决pytorch 编译时CUDA版本与运行时CUDA版本不对应 在跑FPN模型时,发现本机是CUDA10.1,而pytoch==0.4.0,最高支持9.0,故安装CUDA9.0,用cuda-9.0的软连接代替cuda10.1来完成cuda版本的切换(不需要搞cudnn) 替换称cuda9.0后 阅读全文
posted @ 2020-04-01 11:48 Wind·Chaser 阅读(6252) 评论(0) 推荐(1)