tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Thomas Wolf
2024-04-03 07:46:17 +02:00
parent 24821fee24
commit a1e47202c0
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/scripts/eval.py
View File

@@ -271,7 +271,8 @@ if __name__ == "__main__":
elif args.hub_id is not None: elif args.hub_id is not None:
folder = Path(snapshot_download(args.hub_id, revision=args.revision)) folder = Path(snapshot_download(args.hub_id, revision=args.revision))
cfg = init_hydra_config( cfg = init_hydra_config(
folder / "config.yaml", [f"policy.pretrained_model_path={folder / 'model.pt'}", *args.overrides] folder / "config.yaml", [*args.overrides]
# folder / "config.yaml" # , [f"policy.pretrained_model_path={folder / 'model.pt'}", *args.overrides]
) )
stats_path = folder / "stats.pth" stats_path = folder / "stats.pth"