search

HuangJunJie2017 / BEVDet

Code base of the BEVDet series .
Apache License 2.0
1.45k stars 265 forks source link

RuntimeError: shape '[8, 1, 3, 3]' is invalid for input of size 128 #332

Closed VincentZ-2020 closed 8 months ago

VincentZ-2020 commented 8 months ago

能正常训练bevdet-r50-cbgs.py，但是跑带4D前缀的config均出现以下错误

File "/workspace/project/bevdet/BEVDet/mmdet3d/models/necks/view_transformer.py", line 747, in get_mlp_input bda = bda.view(B, 1, 3, 3).repeat(1, N, 1, 1) RuntimeError: shape '[8, 1, 3, 3]' is invalid for input of size 128

历史issue未发现类似的问题，请看看

HuangJunJie2017 commented 8 months ago

@VincentZ-2020 fixed with bda = bda.view(B, 1, 4, 4).repeat(1, N, 1, 1) in https://github.com/HuangJunJie2017/BEVDet/commit/58c2587a8f89a1927926f0bdb6cde2917c91a9a5