DLR-RM/stable-baselines3

How do I export RecurrentPPO as an onnx model？

Open

#1125 aperta il 17 ott 2022

Vedi su GitHub
 (3 commenti) (0 reazioni) (0 assegnatari)Python (1407 fork)batch import
help wantedquestion

Metriche repository

Star
 (6550 star)
Metriche merge PR
 (Merge medio 11g 13h) (3 PR mergiate in 30 g)

Descrizione

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

Guida contributor

Direzione di ricerca
Controlla se stable baselines3 ha utility o esempi di esportazione ONNX esistenti. Guarda l'esportazione ONNX di PyTorch per modelli basati su RNN (usa 'torch.onnx.export' con 'opset version' e assi dinamici). Rivedi la documentazione dell'architettura di RecurrentPPO per capire come tracciare il modello. Considera di usare 'torch.onnx.export' con forme di input e lunghezze di sequenza appropriate.
Tech stack
pythonpytorch
Dominio
machine learning
Tipo issue
Ricerca
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonPyTorchONNX
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
55