RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for ar

运行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))

就没问题了