princeton-vl / RAFT-Stereo

MIT License
686 stars 131 forks source link

Question about --corr_implementation reg cuda option error #97

Open gimunum opened 4 months ago

gimunum commented 4 months ago

When running the demo code in an A6000, ubuntu18.04, python 3.7, cuda 11.8 environment, if I use the reg or alt option as a ' --corr_implementation' option, it runs without error, but if I use the regcuda option, the error below appears. What is the solution? Traceback (most recent call last): File "demo.py", line 78, in demo(args) File "demo.py", line 46, in demo , flow_up = model(image1, image2, iters=args.valid_iters, test_mode=True) File "/home/gmum/anaconda3/envs/raftstereo/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl return forward_call(*input, **kwargs) File "core/raft_stereo.py", line 110, in forward corr = corr_fn(coords1) # index correlation volume File "/home/gmum/PycharmProjects/RAFT-Stereo/core/corr.py", line 50, in call out_pyramid.append(corr.view(bz, -1, ht, wd)) AttributeError: 'list' object has no attribute 'view'