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

View File

@ -271,7 +271,8 @@ if __name__ == "__main__":
elif args.hub_id is not None:
folder = Path(snapshot_download(args.hub_id, revision=args.revision))
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"