DLR-RM/stable-baselines3

How do I export RecurrentPPO as an onnx model？

Open

#1,125 建立於 2022年10月17日

在 GitHub 查看
 (3 留言) (0 反應) (0 負責人)Python (6,550 star) (1,407 fork)batch import
help wantedquestion

描述

❓ Question

How do I export RecurrentPPO as an onnx model？

Checklist

  • I have checked that there is no similar issue in the repo
  • I have read the documentation
  • If code there is, it is minimal and working
  • If code there is, it is formatted using the markdown code blocks for both code and stack traces.

貢獻者指南

技術棧
pythonpytorch
領域
machine learningai
議題類型
documentation
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
2
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
under 1 hour
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
stale
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
clear
前置要求
無
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
70
研究方向
This issue asks how to export a RecurrentPPO model to ONNX format. The contributor should explore the existing ONNX export utilities in the stable baselines3 codebase (e.g., in sb3.common.onnx or similar) and test if they work with recurrent policies. If not, they may need to extend the export function to handle the recurrent state or propose a fix. Finally, they should update the documentation with a step by step guide.