mdaiter · CongHoanCoder · Jan 24, 2026
diff --git 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
diff --git a/outputs/eval/example_aloha-AlohaTransferCube-v0/rollout.mp4 b/outputs/eval/example_aloha-AlohaTransferCube-v0/rollout.mp4
diff --git a/outputs/train/aloha_sim_insertion_human/model_final.safetensors b/outputs/train/aloha_sim_insertion_human/model_final.safetensors
diff --git a/outputs/train/aloha_sim_transfer_cube_human/model_final.safetensors b/outputs/train/aloha_sim_transfer_cube_human/model_final.safetensors
diff --git 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