运行FPN faster rcnn 进行模型继续训练时出现的BUG,RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 ‘other’
查看了网上的原因,发现时数据格式不匹配,必须得改为cuda格式,
所以修改:
if args.resume:
load_name = os.path.join(output_dir,
'fpn_{}_{}_{}.pth'.format(args.checksession, args.checkepoch, args.checkpoint))
在此句之前加上一句转换,变为:
if args.cuda:
FPN.cuda()
if args.resume:
load_name = os.path.join(output_dir,
'fpn_{}_{}_{}.pth'.format(args.checksession, args.checkepoch, args.checkpoint))
就没问题了