search

jfzhang95 / pytorch-video-recognition

PyTorch implemented C3D, R3D, R2Plus1D models for video activity recognition.
MIT License
1.16k stars 250 forks source link

load checkpoint #2

Closed Zhuysheng closed 5 years ago

Zhuysheng commented 5 years ago

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'