b20acd622c
* Update for pyo3 0.21. * Also adapt the RL example. * Fix for the pyo3-onnx bindings... * Print details on failures. * Revert pyi. |
||
|---|---|---|
| .. | ||
| README.md | ||
| atari_wrappers.py | ||
| ddpg.rs | ||
| dqn.rs | ||
| gym_env.rs | ||
| main.rs | ||
| policy_gradient.rs | ||
| vec_gym_env.rs | ||
README.md
candle-reinforcement-learning
Reinforcement Learning examples for candle.
This has been tested with gymnasium version 0.29.1. You can install the
Python package with:
pip install "gymnasium[accept-rom-license]"
In order to run the examples, use the following commands. Note the additional
--package flag to ensure that there is no conflict with the candle-pyo3
crate.
For the Policy Gradient example:
cargo run --example reinforcement-learning --features=pyo3 --package candle-examples -- pg
For the Deep Deterministic Policy Gradient example:
cargo run --example reinforcement-learning --features=pyo3 --package candle-examples -- ddpg