DLR-RM/stable-baselines3

How do I export RecurrentPPO as an onnx model？

Open

#1.125 aberto em 17 de out. de 2022

Ver no GitHub
 (3 comments) (0 reactions) (0 assignees)Python (1.407 forks)batch import
help wantedquestion

Métricas do repositório

Stars
 (6.550 stars)
Métricas de merge de PR
 (Mesclagem média 11d 13h) (3 fundiu PRs em 30d)

Description

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

Guia do colaborador

Direção de pesquisa
Verifique se o stable baselines3 possui utilitários ou exemplos de exportação ONNX existentes. Veja a exportação ONNX do PyTorch para modelos baseados em RNN (use 'torch.onnx.export' com 'opset version' e eixos dinâmicos). Revise a documentação da arquitetura do RecurrentPPO para entender como rastrear o modelo. Considere usar 'torch.onnx.export' com formas de entrada e comprimentos de sequência apropriados.
Pilha de tecnologia
pythonpytorch
Domain
machine learning
Tipo Issue
Pesquisa
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorchONNX
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
55