Saving model to JIT, ONNX

[kemjensak]

HI, I'm using SKRL with Isaac lab
Is there any method to export model with JIT, ONNX format?
RSL-RL has policy exporter as below:
jit, onnx

[Toni-SM]

Hi @kemjensak

Sorry for late response.
Currently no, but it is in the skrl roadmap for a while now, I just haven't had time for it 🙈.
Contributions are welcome! 😅

[renanmb]

I am working on it but I am currently stuck how to make a normalizer like RSL_RL, also the constructors are confusing. The code evolved a lot faster than the docs. https://github.com/leggedrobotics/rsl_rl/blob/main/rsl_rl/modules/normalizer.py#L14

[LemonFoxmere]

Has there been any updates to this?

[Toni-SM]

Hi @LemonFoxmere

There is a PR in progress that implements this feature: #410