search

xiadingZ / video-caption.pytorch

pytorch implementation of video captioning
MIT License
400 stars 128 forks source link

I met dataparallel bug,please help me #43

Closed angela000 closed 4 years ago

angela000 commented 4 years ago

RuntimeError:Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight' ; but device 1 does not equal 0 (while checking arguments for cudnn_convolution)

angela000 commented 4 years ago

change torch version to 0.4.0