search

foolwood / SiamMask

[CVPR2019] Fast Online Object Tracking and Segmentation: A Unifying Approach
http://www.robots.ox.ac.uk/~qwang/SiamMask
MIT License
3.47k stars 816 forks source link

The bug in running demo #170

Open jfkuang opened 4 years ago

jfkuang commented 4 years ago

THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=663 error=11 : invalid argument Traceback (most recent call last): File "/home/jfkuang/SiamMask/tools/demo.py", line 53, in state = siamese_init(im, target_pos, target_sz, siammask, cfg['hp'], device=device) # init tracker File "/home/jfkuang/SiamMask/tools/test.py", line 155, in siamese_init net.template(z.to(device)) File "/home/jfkuang/SiamMask/experiments/siammask_sharp/custom.py", line 174, in template self.zf = self.features(template) File "/home/jfkuang/anaconda3/envs/siammask/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, kwargs) File "/home/jfkuang/SiamMask/experiments/siammask_sharp/custom.py", line 59, in forward output = self.features(x) File "/home/jfkuang/anaconda3/envs/siammask/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, *kwargs) File "/home/jfkuang/SiamMask/experiments/siammask_sharp/resnet.py", line 220, in forward x = self.conv1(x) File "/home/jfkuang/anaconda3/envs/siammask/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(input, kwargs) File "/home/jfkuang/anaconda3/envs/siammask/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward self.padding, self.dilation, self.groups) RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:663

balablb commented 3 years ago

THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=663 error=11 : invalid argument Traceback (most recent call last): File "/home/jfkuang/SiamMask/tools/demo.py", line 53, in state = siamese_init(im, target_pos, target_sz, siammask, cfg['hp'], device=device) # init tracker File "/home/jfkuang/SiamMask/tools/test.py", line 155, in siamese_init net.template(z.to(device)) File "/home/jfkuang/SiamMask/experiments/siammask_sharp/custom.py", line 174, in template self.zf = self.features(template) File "/home/jfkuang/anaconda3/envs/siammask/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, kwargs) File "/home/jfkuang/SiamMask/experiments/siammask_sharp/custom.py", line 59, in forward output = self.features(x) File "/home/jfkuang/anaconda3/envs/siammask/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, kwargs) File "/home/jfkuang/SiamMask/experiments/siammask_sharp/resnet.py", line 220, in forward x = self.conv1(x) File "/home/jfkuang/anaconda3/envs/siammask/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call* result = self.forward(input, kwargs) File "/home/jfkuang/anaconda3/envs/siammask/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward self.padding, self.dilation, self.groups) RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:663

I meet the same error .Have you solved it?

shlsp commented 3 years ago

Annotate torch.backends.cudnn.benchmark = True