hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Refactor configs to have env in seperate yaml + Fix training

Browse Source
This commit is contained in:
Cadene
2024-02-25 17:42:47 +00:00
parent eec134d72b
commit b16c334825
13 changed files with 146 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/utils.py
View File

@@ -4,8 +4,8 @@ from hydra import compose, initialize
CONFIG_PATH = "../lerobot/configs"
def init_config(config_name):
def init_config(config_name="default", overrides=None):
hydra.core.global_hydra.GlobalHydra.instance().clear()
initialize(config_path=CONFIG_PATH)
cfg = compose(config_name=config_name)
cfg = compose(config_name=config_name, overrides=overrides)
return cfg