DLR-RM/stable-baselines3

How do I export RecurrentPPO as an onnx model？

開放

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

在 GitHub 查看貢獻者指南
 (3 則留言) (0 個反應) (0 位負責人)Python (1,407 個分叉)batch import
help wantedquestion

倉庫指標

星標
 (6,550 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

❓ 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.

貢獻者指南

研究方向
檢查stable baselines3是否有現有的ONNX匯出工具或範例。查看PyTorch對基於RNN的模型的ONNX匯出（使用'torch.onnx.export'，注意'opset version'和動態軸）。查閱RecurrentPPO的架構文件以了解如何追蹤模型。考慮使用'torch.onnx.export'並指定適當的輸入形狀和序列長度。
技術棧
pythonpytorch
領域
machine learning
議題類型
調研
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-3 小時
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
PythonPyTorchONNX
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
55