search

hyseob / PyMDNet

MDNet PyTorch implementation
Other
420 stars 150 forks source link

cuda error (python run_tracker.py) #23

Closed dhlee-jubilee closed 5 years ago

dhlee-jubilee commented 5 years ago

I got some issue with inference step. I think it's pytorch + cuda issue. what's the problem?

python run_tracker.py -s DragonBaby -f

File "run_tracker.py", line 329, in result, result_bb, fps = run_mdnet(img_list, init_bbox, gt=gt, savefig_dir=savefig_dir, display=display) File "run_tracker.py", line 131, in run_mdnet model = model.cuda() File "/home/dhlee/anaconda2/envs/tensorflow/lib/python2.7/site-packages/torch/nn/modules/module.py", line 147, in cuda return self._apply(lambda t: t.cuda(device_id)) File "/home/dhlee/anaconda2/envs/tensorflow/lib/python2.7/site-packages/torch/nn/modules/module.py", line 118, in _apply module._apply(fn) File "/home/dhlee/anaconda2/envs/tensorflow/lib/python2.7/site-packages/torch/nn/modules/module.py", line 118, in _apply module._apply(fn) File "/home/dhlee/anaconda2/envs/tensorflow/lib/python2.7/site-packages/torch/nn/modules/module.py", line 118, in _apply module._apply(fn) File "/home/dhlee/anaconda2/envs/tensorflow/lib/python2.7/site-packages/torch/nn/modules/module.py", line 124, in _apply param.data = fn(param.data) File "/home/dhlee/anaconda2/envs/tensorflow/lib/python2.7/site-packages/torch/nn/modules/module.py", line 147, in return self._apply(lambda t: t.cuda(device_id)) File "/home/dhlee/anaconda2/envs/tensorflow/lib/python2.7/site-packages/torch/_utils.py", line 65, in _cuda return newtype(self.size()).copy(self, async) File "/home/dhlee/anaconda2/envs/tensorflow/lib/python2.7/site-packages/torch/cuda/init.py", line 275, in new return super(_CudaBase, cls).new(cls, *args, **kwargs) KeyboardInterrupt