From 094ee0c5ba39e5a5612ebdf9752b5550fead5092 Mon Sep 17 00:00:00 2001 From: Axel Brunnbauer Date: Tue, 15 Jul 2025 23:02:04 -0700 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index b3f3de8..ad66f89 100644 --- a/README.md +++ b/README.md @@ -36,7 +36,7 @@ pip install -e . ## Running Experiments -The main code for the algorithm is in `src/reppo_jax/reppo.py` and `src/torchrl/reppo.py` respectively. +The main code for the algorithm is in `src/jaxrl/reppo.py` and `src/torchrl/reppo.py` respectively. In our tests, both versions produce similar returns up to seed variance. However, due to slight variations in the frameworks, we cannot always guarantee this. @@ -46,7 +46,7 @@ This can result in cases where the GPU is stalled if the CPU cannot provide inst Our configurations are handled with [hydra.cc](https://hydra.cc/). This means parameters can be overwritten by using the syntax ```bash -python src/reppo_jax/reppo.py PARAMETER=VALUE +python src/jaxrl/reppo.py PARAMETER=VALUE ``` By default, the environment type and name need to be provided.