摘要:
https://www.jianshu.com/p/54996f2808e1 比较详细的FPN教程,参考 阅读全文
posted @ 2020-04-01 23:42
Wind·Chaser
阅读(2422)
评论(0)
推荐(0)
RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for ar
摘要:
这行报错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)

浙公网安备 33010602011771号