ready for review

This commit is contained in:
Alexander Soare
2024-04-16 13:43:58 +01:00
parent 03b08eb74e
commit 9c2f10bd04
4 changed files with 26 additions and 16 deletions

View File

@@ -11,6 +11,7 @@ from lerobot.common.utils import init_hydra_config
from lerobot.scripts.eval import eval
# Get a pretrained policy from the hub.
# TODO(alexander-soare): This no longer works until we upload a new model that uses the current configs.
hub_id = "lerobot/diffusion_policy_pusht_image"
folder = Path(snapshot_download(hub_id))
# OR uncomment the following to evaluate a policy from the local outputs/train folder.