Skip to content

Commit

Permalink
update README and config
Browse files Browse the repository at this point in the history
  • Loading branch information
M. Choi committed Jul 8, 2024
1 parent 23a9c47 commit 400de2a
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 21 deletions.
15 changes: 14 additions & 1 deletion examples/sweep/wandb/readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,21 @@ You can improve the performance of your RL agents with hyperparameter tuning. It

## Prerequisites

Ensure you have Weights & Biases: `pip install wandb` installed on top of benchmarl requirements.
- Ensure you have Weights & Biases: `pip install wandb` installed on top of benchmarl requirements.

- Update the `benchmarl/conf/config.yaml` with your desired experiment setup, e.g.:

```yaml
defaults:
- experiment: base_experiment
- algorithm: ippo
- task: customenv/task_1
- model: layers/mlp
- model@critic_model: layers/mlp
- _self_

seed: 0
```
## Step 1: Define Your Sweep Configuration
Expand Down
21 changes: 1 addition & 20 deletions examples/sweep/wandb/sweepconfig.yaml
Original file line number Diff line number Diff line change
@@ -1,23 +1,5 @@
'''
Read this blog post for more details
https://wandb.ai/adrishd/hydra-example/reports/Configuring-W-B-Projects-with-Hydra--VmlldzoxNTA2MzQw?galleryTag=posts&utm_source=fully_connected&utm_medium=blog&utm_campaign=hydra
Also you need to change the config.yaml like following:
# defaults:
# - experiment: base_experiment
# - algorithm: ippo
# - task: customenv/task_1
# - model: layers/mlp
# - model@critic_model: layers/mlp
# - _self_
# seed: 0
'''

program: PATH_TO_YOUR_DIRECTORY\benchmarl\run.py
project: "sweep_benchmarl"
project: "YOUR_PROJECT_NAME"
entity: "YOUR_ENTITY_NAME"

method: bayes
Expand Down Expand Up @@ -57,7 +39,6 @@ parameters:
# distribution: categorical



early_terminate:
type: hyperband
max_iter: 27
Expand Down

0 comments on commit 400de2a

Please sign in to comment.