mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00

Files

Travis Hammond 498c50348c Add DDPG and fix Gym wrapper (#1207 )

* Fix Gym wrapper
- It was returning things in the wrong order
- Gym now differentiates between terminated and truncated

* Add DDPG

* Apply fixes

* Remove Result annotations

* Also remove Vec annotation

* rustfmt

* Various small improvements (avoid cloning, mutability, get clippy to pass, ...)

---------

Co-authored-by: Travis Hammond <travis.hammond@alexanderthamm.com>
Co-authored-by: Laurent <laurent.mazare@gmail.com>

2023-10-28 19:53:34 +01:00

atari_wrappers.py

Add some reinforcement learning example. (#1090 )

2023-10-14 16:46:43 +01:00

ddpg.rs

Add DDPG and fix Gym wrapper (#1207 )

2023-10-28 19:53:34 +01:00

gym_env.rs

Add DDPG and fix Gym wrapper (#1207 )

2023-10-28 19:53:34 +01:00

main.rs

Add DDPG and fix Gym wrapper (#1207 )

2023-10-28 19:53:34 +01:00

README.md

Add some reinforcement learning example. (#1090 )

2023-10-14 16:46:43 +01:00

vec_gym_env.rs

Add some reinforcement learning example. (#1090 )

2023-10-14 16:46:43 +01:00

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