Tracker: ONNX Model Exporting Progress

edbeeching / godot_rl_agents

An Open Source package that allows video game creators, AI researchers and hobbyists the opportunity to learn complex behaviors for their Non Player Characters or agents

MIT License

909 stars 64 forks source link

Tracker: ONNX Model Exporting Progress #84

Open yaelatletl opened 1 year ago

yaelatletl commented 1 year ago

Currently, RL-lib agents can be exported to ONNX files, but there are other backends which haven't recieved export support.

[x] StableBaselines3
[ ] Sample Factory
[x] Ray RLLib
[ ] CleanRL

edbeeching commented 1 year ago

I looked for SB3, it seems that broadcast tensors are not supported out of the box, but there is a workaround.

yaelatletl commented 1 year ago

Added tracker from #105