Add VQ-BeT (#166)

This commit is contained in:
Seungjae Lee
2024-06-26 16:55:02 +09:00
committed by GitHub
parent 9aa4cdb976
commit 7d1542cae1
9 changed files with 2644 additions and 1 deletions

View File

@@ -134,12 +134,13 @@ available_policies = [
"act",
"diffusion",
"tdmpc",
"vqbet",
]
# keys and values refer to yaml files
available_policies_per_env = {
"aloha": ["act"],
"pusht": ["diffusion"],
"pusht": ["diffusion", "vqbet"],
"xarm": ["tdmpc"],
"dora_aloha_real": ["act_real"],
}