From 53ff3544bf65177b07273926e32cc4ae39029fa4 Mon Sep 17 00:00:00 2001 From: Cong Hoan Nguyen Date: Sat, 24 Jan 2026 17:34:34 -0500 Subject: [PATCH] re-run and test --- .gitignore | 2 +- outputs/eval/example_aloha-AlohaInsertion-v0/rollout.mp4 | 3 +++ outputs/eval/example_aloha-AlohaTransferCube-v0/rollout.mp4 | 3 +++ .../train/aloha_sim_insertion_human/model_final.safetensors | 3 +++ .../aloha_sim_transfer_cube_human/model_final.safetensors | 3 +++ test.py | 5 +++-- 6 files changed, 16 insertions(+), 3 deletions(-) create mode 100644 outputs/eval/example_aloha-AlohaInsertion-v0/rollout.mp4 create mode 100644 outputs/eval/example_aloha-AlohaTransferCube-v0/rollout.mp4 create mode 100644 outputs/train/aloha_sim_insertion_human/model_final.safetensors create mode 100644 outputs/train/aloha_sim_transfer_cube_human/model_final.safetensors diff --git a/.gitignore b/.gitignore index 81c0f48..372c13e 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,2 @@ __pycache__/ -outputs/ + diff --git a/outputs/eval/example_aloha-AlohaInsertion-v0/rollout.mp4 b/outputs/eval/example_aloha-AlohaInsertion-v0/rollout.mp4 new file mode 100644 index 0000000..0fdda71 --- /dev/null +++ b/outputs/eval/example_aloha-AlohaInsertion-v0/rollout.mp4 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:84fd3884c53fb19e2a6f0dda8b2bf5d1953488e48008077157a416a71c841fb3 +size 82674 diff --git a/outputs/eval/example_aloha-AlohaTransferCube-v0/rollout.mp4 b/outputs/eval/example_aloha-AlohaTransferCube-v0/rollout.mp4 new file mode 100644 index 0000000..f8b9b1c --- /dev/null +++ b/outputs/eval/example_aloha-AlohaTransferCube-v0/rollout.mp4 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2524f2db3fd73f356fde696fa593e6f0586c27c0a6df227502b5e6f65e3dda02 +size 84098 diff --git a/outputs/train/aloha_sim_insertion_human/model_final.safetensors b/outputs/train/aloha_sim_insertion_human/model_final.safetensors new file mode 100644 index 0000000..60d3032 --- /dev/null +++ b/outputs/train/aloha_sim_insertion_human/model_final.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:46379077477b28df4c022404be8e3b3e40a4741983c83085fba67f2afad499aa +size 206771192 diff --git a/outputs/train/aloha_sim_transfer_cube_human/model_final.safetensors b/outputs/train/aloha_sim_transfer_cube_human/model_final.safetensors new file mode 100644 index 0000000..dfcfa26 --- /dev/null +++ b/outputs/train/aloha_sim_transfer_cube_human/model_final.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fbc0c710cc872f89a4ad80aa05d2f5b58871818386e1e0202ef056b70159332e +size 206771192 diff --git a/test.py b/test.py index 3c9cac3..d10c94d 100644 --- a/test.py +++ b/test.py @@ -16,8 +16,9 @@ import argparse parser=argparse.ArgumentParser(description="Argument Parser for ACT testing on simulated environments") -parser.add_argument("env_name", type=str, choices=['AlohaTransferCube-v0', 'AlohaInsertion-v0'], default='AlohaTransferCube-v0') -parser.add_argument("model_path", type=str) +parser.add_argument("--env_name", type=str, choices=['AlohaTransferCube-v0', 'AlohaInsertion-v0'], default='AlohaInsertion-v0') +parser.add_argument("--model_path", type=str, default='outputs/train/aloha_sim_insertion_human/model_50000.safetensors') +# parser.add_argument("--model_path", type=str, default='outputs/train/aloha_sim_transfer_cube_human/model_30000.safetensors') args=parser.parse_args() env_name = args.env_name