summaryrefslogtreecommitdiff
path: root/candle-examples/examples/reinforcement-learning/README.md
diff options
context:
space:
mode:
authorIvar Flakstad <69173633+ivarflakstad@users.noreply.github.com>2024-01-07 11:52:03 +0100
committerIvar Flakstad <69173633+ivarflakstad@users.noreply.github.com>2024-01-07 11:52:03 +0100
commit6ebe04327397ebf7c9400d68c43fede705f8ce75 (patch)
tree4e701ef55370cb6226593019f9e6daaf8ddf73b4 /candle-examples/examples/reinforcement-learning/README.md
parent6bf52b9fdf82ad775611e82924d73172660a605e (diff)
parent84250bf52f58528cf59dca3b82effd9f07a13cc7 (diff)
downloadcandle-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.md11
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
```