uber-research / DeepPruner

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

submission_kitti.py #38

Closed zpp1996 closed 2 years ago

zpp1996 commented 2 years ago

I set args.cuda=False .But RuntimeError: Device index must not be negative This error occurred in deeppruner.py device = left_input.get_device() if stage is "pre": min_disparity = torch.zeros((left_input.size()[0], 1, left_input.size()[2], left_input.size()[3]), device=device) max_disparity = torch.zeros((left_input.size()[0], 1, left_input.size()[2], left_input.size()[3]), device=device) + self.max_disp