Add VQ-BeT (#166)

2024-06-26 16:55:02 +09:00
parent 9aa4cdb976
commit 7d1542cae1
9 changed files with 2644 additions and 1 deletions
--- a/tests/test_policies.py
+++ b/tests/test_policies.py
@@ -49,6 +49,7 @@ def test_get_policy_and_config_classes(policy_name: str):
    [
        ("xarm", "tdmpc", ["policy.use_mpc=true", "dataset_repo_id=lerobot/xarm_lift_medium"]),
        ("pusht", "diffusion", []),
+        ("pusht", "vqbet", []),
        ("aloha", "act", ["env.task=AlohaInsertion-v0", "dataset_repo_id=lerobot/aloha_sim_insertion_human"]),
        (
            "aloha",