From 440cc84131cd75e7dd99fe59ca47d4dbf7c4f674 Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Fri, 31 May 2024 11:42:06 +0000 Subject: [PATCH] fixed --- .../dora_aloha_real_act_real/actions.safetensors | 4 ++-- .../dora_aloha_real_act_real_no_state/actions.safetensors | 4 ++-- tests/scripts/save_policy_to_safetensors.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/actions.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/actions.safetensors index 9f3d6dae..2373f1ee 100644 --- a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/actions.safetensors +++ b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/actions.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:218f3f0317515aa95a1aa8ab0fcc95300466ec25cd44546d68a67b730aa0d2d0 -size 51248 +oid sha256:2fff6294b94cf42d4dd1249dcc5c3b0269d6d9c697f894e61b867d7ab81a94e4 +size 5104 diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/actions.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/actions.safetensors index b09080d9..9f0ba883 100644 --- a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/actions.safetensors +++ b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/actions.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:77788857dc956bee08b61b932da6277fa386cac38dbe28acc35fef3125f5ed6a -size 51248 +oid sha256:fb1a45463efd860af2ca22c16c77d55a18bd96fef080ae77978845a2f22ef716 +size 5104 diff --git a/tests/scripts/save_policy_to_safetensors.py b/tests/scripts/save_policy_to_safetensors.py index c8c0b6cd..961b7cef 100644 --- a/tests/scripts/save_policy_to_safetensors.py +++ b/tests/scripts/save_policy_to_safetensors.py @@ -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)