median-research-group / LibMTL

A PyTorch Library for Multi-Task Learning
MIT License
1.94k stars 181 forks source link

How to export saved models to other formats, such as onnx, mnn, etc #51

Closed zcc720 closed 1 year ago

zcc720 commented 1 year ago

Hello, first of all, thank you for your great work. I saved the best model while running the Office example, but I want to export this model as an ONNX model and use tools like Netron to understand the model structure. How can I add code to the framework? Can you add an exported component for exporting various models, such as onnx, mnn, tflite

Baijiong-Lin commented 1 year ago

Sorry, I am not familiar with ONNX. But LibMTL supports saving model now. Can't it be converted to ONNX like other torch models?

Baijiong-Lin commented 1 year ago

Closed as no further updates.