diff options
author | Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-07 11:52:03 +0100 |
---|---|---|
committer | Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-07 11:52:03 +0100 |
commit | 6ebe04327397ebf7c9400d68c43fede705f8ce75 (patch) | |
tree | 4e701ef55370cb6226593019f9e6daaf8ddf73b4 /candle-examples/examples/reinforcement-learning/README.md | |
parent | 6bf52b9fdf82ad775611e82924d73172660a605e (diff) | |
parent | 84250bf52f58528cf59dca3b82effd9f07a13cc7 (diff) | |
download | candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.tar.gz candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.tar.bz2 candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.zip |
Merge branch 'main' into ivarflakstad/metal-prng
Diffstat (limited to 'candle-examples/examples/reinforcement-learning/README.md')
-rw-r--r-- | candle-examples/examples/reinforcement-learning/README.md | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/candle-examples/examples/reinforcement-learning/README.md b/candle-examples/examples/reinforcement-learning/README.md index 2d3d14b0..28819067 100644 --- a/candle-examples/examples/reinforcement-learning/README.md +++ b/candle-examples/examples/reinforcement-learning/README.md @@ -8,9 +8,16 @@ Python package with: pip install "gymnasium[accept-rom-license]" ``` -In order to run the example, use the following command. Note the additional +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: +```bash +cargo run --example reinforcement-learning --features=pyo3 --package candle-examples -- pg +``` + +For the Deep Deterministic Policy Gradient example: ```bash -cargo run --example reinforcement-learning --features=pyo3 --package candle-examples +cargo run --example reinforcement-learning --features=pyo3 --package candle-examples -- ddpg ``` |