mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-21 03:30:10 +00:00
add franka action
This commit is contained in:
+2
-3
@@ -1,6 +1,5 @@
|
||||
from lerobot.policies.factory import make_policy
|
||||
from lerobot.policies.factory import make_policy_config
|
||||
from lerobot.datasets.lerobot_dataset import LeRobotDatasetMetadata
|
||||
from lerobot.policies.factory import make_policy, make_policy_config
|
||||
|
||||
cfg = make_policy_config("xvla")
|
||||
|
||||
@@ -8,4 +7,4 @@ dataset_id = "lerobot/svla_so101_pickplace"
|
||||
# This only downloads the metadata for the dataset, ~10s of MB even for large-scale datasets
|
||||
dataset_metadata = LeRobotDatasetMetadata(dataset_id)
|
||||
policy = make_policy(cfg=cfg, ds_meta=dataset_metadata)
|
||||
print(policy)
|
||||
print(policy)
|
||||
|
||||
Reference in New Issue
Block a user