add h5out as arg to produce_sim_data as well

kscalelabs · Nov 20, 2024 · 6a65d01 · 6a65d01
1 parent 833f03a
commit 6a65d01
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sim/produce_sim_data.py b/sim/produce_sim_data.py
@@ -17,6 +17,8 @@ def run_simulation(sim_idx: int, args: argparse.Namespace) -> None:
         "--load_model",
         args.load_model,
         "--log_h5",
+        "--h5_out_dir",
+        args.h5_out_dir,
         "--no_render",
     ]
 
@@ -48,6 +50,7 @@ def run_parallel_sims(num_threads: int, args: argparse.Namespace) -> None:
     parser.add_argument("--embodiment", default="stompypro", type=str, help="Embodiment name")
     parser.add_argument("--load_model", default="examples/walking_pro.onnx", type=str, help="Path to model to load")
     parser.add_argument("--num_examples", default=10000, type=int, help="Number of examples to run")
+    parser.add_argument("--h5_out_dir", default="sim/resources", type=str, help="Directory to save HDF5 files")
     args = parser.parse_args()
 
     # Run 100 examples total, in parallel batches