wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixed

This commit is contained in:
Remi Cadene 2024-05-31 11:42:06 +00:00
parent 144f723c16
commit 440cc84131
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/actions.safetensors
View File

@ -1,3 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:218f3f0317515aa95a1aa8ab0fcc95300466ec25cd44546d68a67b730aa0d2d0
size 51248
oid sha256:2fff6294b94cf42d4dd1249dcc5c3b0269d6d9c697f894e61b867d7ab81a94e4
size 5104

4
tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/actions.safetensors
View File

@ -1,3 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:77788857dc956bee08b61b932da6277fa386cac38dbe28acc35fef3125f5ed6a
size 51248
oid sha256:fb1a45463efd860af2ca22c16c77d55a18bd96fef080ae77978845a2f22ef716
size 5104

4
tests/scripts/save_policy_to_safetensors.py
View File

@ -115,8 +115,8 @@ if __name__ == "__main__":
["policy.n_action_steps=8", "policy.num_inference_steps=10", "policy.down_dims=[128, 256, 512]"],
),
("aloha", "act", ["policy.n_action_steps=10"]),
("dora_aloha_real", "act_real", []),
("dora_aloha_real", "act_real_no_state", []),
("dora_aloha_real", "act_real", ["policy.n_action_steps=10"]),
("dora_aloha_real", "act_real_no_state", ["policy.n_action_steps=10"]),
]
for env, policy, extra_overrides in env_policies:
save_policy_to_safetensors("tests/data/save_policy_to_safetensors", env, policy, extra_overrides)