Init

2025-09-07 14:59:17 +08:00
commit 24dbdbd24b
40 changed files with 10754 additions and 0 deletions
@@ -0,0 +1,24 @@
+#!/bin/bash
+# export CUDA_VISIBLE_DEVICES=4,5,6,7
+export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
+NUM_GPUS=$(echo $CUDA_VISIBLE_DEVICES | tr ',' '\n' | wc -l)
+
+# print current time
+echo "[current time: $(date +'%Y-%m-%d %H:%M:%S')]"
+
+code_dir="/path/to/wall-x"
+config_path="/path/to/wall-x/workspace/lerobot_example"
+
+# Use a fixed port instead of a random one
+export PORT=$((21000 + $RANDOM % 30000))
+
+MASTER_PORT=10239 # use 5 digits ports
+
+export LAUNCHER="accelerate launch --num_processes=$NUM_GPUS --main_process_port=$PORT"
+
+export SCRIPT="${code_dir}/train_qact.py"
+export SCRIPT_ARGS="--config ${config_path}/config_qact.yml --seed $MASTER_PORT"
+
+echo "Running command: $LAUNCHER $SCRIPT $SCRIPT_ARGS"
+
+$LAUNCHER $SCRIPT $SCRIPT_ARGS