search

lix19937 / tensorrt-insight

deep insight tensorrt
1 stars 0 forks source link

squeeze(-1) 可能导致if 操作产生 可通过onnx 看到 #15

Open lix19937 opened 4 weeks ago

lix19937 commented 4 weeks ago 
# bev_mask = bev_mask.permute(2, 1, 3, 0, 4).squeeze(-1)
  bev_mask = bev_mask.permute(2, 1, 3, 0, 4)[..., 0]
lix19937 commented 4 weeks ago

问题产生于 BEVFormer-master/projects/mmdet3d_plugin/bevformer/modules/encoder.py