huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0d9bb4eb18b01a06975e2d6eed2cec6735ec8400
candle/candle-examples/examples/reinforcement-learning/README.md
Laurent Mazare 29c7f2565d Add some reinforcement learning example. (#1090) 
* Add some reinforcement learning example.

* Python initialization.

* Get the example to run.

* Vectorized gym envs for the atari wrappers.

* Get some simulation loop to run.
2023-10-14 16:46:43 +01:00

488 B
Raw Blame History

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 example, use the following command. Note the additional --package flag to ensure that there is no conflict with the candle-pyo3 crate.

cargo run --example reinforcement-learning --features=pyo3 --package candle-examples