diff --git a/README.md b/README.md index d71fad79..0786c6d6 100644 --- a/README.md +++ b/README.md @@ -141,7 +141,6 @@ Or you can achieve the same result by executing our script from the command line ```bash python lerobot/scripts/eval.py \ --hub-id lerobot/diffusion_policy_pusht_image \ ---revision v1.0 \ eval_episodes=10 \ hydra.run.dir=outputs/eval/example_hub ``` diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index 72517504..2a3ab13b 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -249,7 +249,7 @@ if __name__ == "__main__": # TODO(alexander-soare): Save and load stats in trained model directory. stats_path = None elif args.hub_id is not None: - folder = Path(snapshot_download(args.hub_id, revision="v1.0")) + 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] )