Model preservation and transformation problems(模型保存和转换的问题)

[CvBokchoy]

How do I save the model trained with the example as a pt file or something else and convert it to an onnx model。
（请问如何将用示例训练好的模型保存为pt文件或者其他，并且转换为onnx模型？？）

[josef821]

hi, i use this code to my unet;
after running my model with this code:
outputs = model(IMGDATA)
i use this code for export to onnx
torch.onnx.export(model,IMGDATA,"model.onnx")

[CvBokchoy]

hi, i use this code to my unet; after running my model with this code: outputs = model(IMGDATA) i use this code for export to onnx torch.onnx.export(model,IMGDATA,"model.onnx")

thank you very much！

[CvBokchoy]

hi, i use this code to my unet; after running my model with this code: outputs = model(IMGDATA) i use this code for export to onnx torch.onnx.export(model,IMGDATA,"model.onnx")

Could you share your C++ tensorRT pre-deployment and post-deployment code snippets, I'm having a little trouble here. thank you

[josef821]

im not coding in c++

[CvBokchoy]

im not coding in c++

Can you help me? i need a C++ tensorrt version inference this model sample. thank you very much!

[CvBokchoy]

im not coding in c++

Can you help me? i need a C++ tensorrt version inference this model sample. thank you very much!

[josef821]

i don't have any source code of c++ tensorrt.