uber-research / DeepPruner

DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)
Other
354 stars 41 forks source link

about submission_kitti.py #11

Closed YongYongLi123 closed 4 years ago

YongYongLi123 commented 4 years ago

Hello: When I run submission_kitti.py I get RuntimeError: cuda runtime error (11): invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:663, what should I do, thanks

ShivamDuggal4 commented 4 years ago

Hi @YongYongLi123 This is probably some issue with your CUDA setup. Check this link.

You can set args.cuda parameter as False, in case you do not want to use it.

Closing this issue on this note, feel free to reopen in case of any further issue.