Exporting burn model.

[wangjiawen2013]

Hi,
How to export burn models so that we can use it in other frameworks such as pytorch ?

[HessTaha]

Hello,

I believe you can export the model in ONNX from there you Can load it back using pytorch?

PS: I've never tried that

[laggui]

We support ONNX import but not export.

There is no official way to export to another format, but if you save it with one of the json file recorders you could load the weights in python with your framework of choice.

For pytorch you'll probably need to map the key-value pairs to the correct state_dict based on your model definition.

P.S: questions of this nature are usually better suited for discussions 🙂

[antimora]

The default file format is in message pack which is portable and supported by many languages. JSON also works but it could be large.