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

Rename Diffusion Policy model repository (#136)

Browse Source
This commit is contained in:
Alexander Soare
2024-05-05 12:06:58 +01:00
committed by GitHub
parent f3bba0270d
commit f6fe162432
4 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/2_evaluate_pretrained_policy.py
View File

@@ -10,7 +10,7 @@ from huggingface_hub import snapshot_download
from lerobot.scripts.eval import eval
# Get a pretrained policy from the hub.
pretrained_policy_name = "lerobot/diffusion_policy_pusht_image"
pretrained_policy_name = "lerobot/diffusion_pusht"
pretrained_policy_path = Path(snapshot_download(pretrained_policy_name))
# OR uncomment the following to evaluate a policy from the local outputs/train folder.
# pretrained_policy_path = Path("outputs/train/example_pusht_diffusion")