There was a bug when I continued to train the model. RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'other'

[smileSJin]

when i try to resume to train the model ：
RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'other'

[smileSJin]

I have sloved it，
add this：
if args.cuda:
FPN.cuda()
before the code（about line 290）：
if args.resume:
load_name = os.path.join(output_dir,
'fpn_{}{}{}.pth'.format(args.checksession, args.checkepoch, args.checkpoint))