Onnx export in colab example (Denys88#202)

* updated notebooks * examples now work in onnx Co-authored-by: Denys Makoviichuk <[email protected]>
kellyguo11 · Sep 18, 2022 · 48f6629 · 48f6629
1 parent bc7cc36
commit 48f6629
Show file tree

Hide file tree

Showing 5 changed files with 665 additions and 720 deletions.
diff --git a/README.md b/README.md
@@ -48,6 +48,8 @@ Explore RL Games quick and easily in colab notebooks:
 
 * [Mujoco training](https://colab.research.google.com/github/Denys88/rl_games/blob/master/notebooks/mujoco_envpool_training.ipynb) Mujoco envpool training example.
 * [Brax training](https://colab.research.google.com/github/Denys88/rl_games/blob/master/notebooks/brax_training.ipynb) Brax training example, with keeping all the observations and actions on GPU.
+* [Onnx discrete space export example with Cartpole](https://colab.research.google.com/github/Denys88/rl_games/blob/master/notebooks/train_and_export_onnx_example_discrete.ipynb) Mujoco envpool training example.
+* [Onnx continuous space export example with Pendulum](https://colab.research.google.com/github/Denys88/rl_games/blob/master/notebooks/train_and_export_onnx_example_continuous.ipynb) Brax training example, with keeping all the observations and actions on GPU.
 
 ## Installation
 

diff --git a/notebooks/train_and_export_onnx_example.ipynb b/notebooks/train_and_export_onnx_example.ipynb