how to convert pth or pth.tar to onnx？

talebolano / TensorRT_Seg_LW_Refinenet

一个简单的TensorRT实现语义分割的C++程序

58 stars 18 forks source link

how to convert pth or pth.tar to onnx？ #8

Open brucelee78 opened 4 years ago

brucelee78 commented 4 years ago

Thank you for your open source ! so if i train a model use Light-Weight RefineNet, so i want to know how to convert it to onnx, could you give the script? thank you !!