train from resume epochs

[fish8sank]

optimizer.step()
File "/home/z/anaconda3/envs/py3/lib/python3.6/site-packages/torch/optim/sgd.py", line 101, in step
buf.mul_(momentum).add_(1 - dampening, d_p)
RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'other'

something wrong?

[Jambalaya11]

I met the same problem with you, do you solve the problem yet?

[zugexiaodui]

I met the same problem with you, do you solve the problem yet?

[JinXiaozhao]

I met the same problem with you, do you solve the problem yet? @fish8sank @Jambalaya11 @zugexiaodui @jfzhang95

[zugexiaodui]

I have forgotten how I solved the problem. Maybe it's due to different versions of pytorch.