search

SmallMunich / nutonomy_pointpillars

Convert pointpillars Pytorch Model To ONNX for TensorRT Inference
MIT License
376 stars 85 forks source link

ONXX format #3

Closed MaxandreOgeret closed 4 years ago

MaxandreOgeret commented 4 years ago

Hello, and thanks a lot for your work, it was really helpful !

I can't manage to export the ONXX file in a version that is suitable for my use. I need ONXX v3 and your program is exporting v4.

Do you know how I could change this ? Thanks ! :)

SmallMunich commented 4 years ago

onnx v3 export it well. you re-train model and export it with your onnx version.

MaxandreOgeret commented 4 years ago

Thanks for your answer !

Should I downgrade my Tensorrt version or change the torch.onnx.export function parameters in train.py ?

SmallMunich commented 4 years ago

Thanks for your answer !

Should I downgrade my Tensorrt version or change the torch.onnx.export function parameters in train.py ?

change TensorRT version maybe better.