how to generate model.pth to model.onnx?

aharley / simple_bev

A Simple Baseline for BEV Perception

MIT License

483 stars 75 forks source link

how to generate model.pth to model.onnx? #56

Open zmtttt opened 4 months ago

zmtttt commented 4 months ago

how to get the architechture of the model

zmtttt commented 4 months ago

result = self.forward(*input, **kwargs) TypeError: Segnet.forward() missing 2 required positional arguments: 'pix_T_cams' and 'cam0_T_camXs' why？？ my_onnx_inputs include those, torch.Size([4, 6, 3, 900, 1600]) torch.Size([4, 6, 4, 4]) torch.Size([4, 6, 4, 4]) torch.Size([4, 1, 200, 8, 200])