inference error

[phamhieutn44]

Im using pretrained model with template data and corresponding .pkl file with this: ./tools/dist_inference.sh ./projects/configs/surroundocc/surroundocc_inference.py ./path/to/ckpts.pth 1 Why i got this error

[unuan12]

我是修改的这里，不使用self.is_vis，否则会返回shape[0]，就会遇到问题。

[weiyithu]

不应该出现这个问题啊，如果返回的shape[0]，应该不会执行这段code

[doublekuk]

我是修改的这里，不使用self.is_vis，否则会返回shape[0]，就会遇到问题。

我修改了这个红框里的return pred_occ.shape[0]为return pred_occ，然后模型可以正常预测了