Unverified Commit 9655e620 authored by ooctipus's avatar ooctipus Committed by GitHub

Adds env_cfg and agent_cfg to wandb in rl_games (#2771)

# Description

This PR logs env_cfg and agent_cfg to wandb in accordance how rl_games
handles env_cfg and agent_cfg. Before env_cfg and agent_cfg is not
observable from wandb run.


## Type of change

- New feature (non-breaking change which adds functionality)

## Screenshots

Please attach before and after screenshots of the change if applicable.

<!--
Example:

| Before | After |
| ------ | ----- |
| _gif/png before_ | _gif/png after_ |

To upload images to a PR -- simply drag and drop an image while in edit
mode and it should upload the image directly. You can then paste that
source into the above before/after sections.
-->

## Checklist

- [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with
`./isaaclab.sh --format`
- [ ] I have made corresponding changes to the documentation
- [x] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [x] I have updated the changelog and the corresponding version in the
extension's `config/extension.toml` file
- [x] I have added my name to the `CONTRIBUTORS.md` or my name already
exists there

<!--
As you go through the checklist above, you can mark something as done by
putting an x character in it

For example,
- [x] I have done this task
- [ ] I have not done this task
-->
parent 887342ae
...@@ -194,10 +194,11 @@ def main(env_cfg: ManagerBasedRLEnvCfg | DirectRLEnvCfg | DirectMARLEnvCfg, agen ...@@ -194,10 +194,11 @@ def main(env_cfg: ManagerBasedRLEnvCfg | DirectRLEnvCfg | DirectMARLEnvCfg, agen
entity=args_cli.wandb_entity, entity=args_cli.wandb_entity,
name=experiment_name, name=experiment_name,
sync_tensorboard=True, sync_tensorboard=True,
config=agent_cfg,
monitor_gym=True, monitor_gym=True,
save_code=True, save_code=True,
) )
wandb.config.update({"env_cfg": env_cfg.to_dict()})
wandb.config.update({"agent_cfg": agent_cfg})
if args_cli.checkpoint is not None: if args_cli.checkpoint is not None:
runner.run({"train": True, "play": False, "sigma": train_sigma, "checkpoint": resume_path}) runner.run({"train": True, "play": False, "sigma": train_sigma, "checkpoint": resume_path})
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment