Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
felimomo authored Feb 26, 2024
1 parent b95dfec commit 2d34a98
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ Coming soon: a more detailed explanation of the integral projection model used!

# Training

We use YAML files to specify training instances, examples of these files can be found on `hyperpars`.
Key components there are: specifying the RL algorithm (e.g. PPO or TQC), specifying the environment trained on (e.g. `GreenCrab-v2`) and the training duration (6M time steps in our examples).
To train using this framework, use the command `python scripts/train.py path/to/file.yml`.
`python scripts/train.py -f ../hyperpars/ppo-gcse.yml`

Alternatively, `bash scripts/train_algos.sh` trains using several algorithms in parallel.

0 comments on commit 2d34a98

Please sign in to comment.