Update iterate_rollout.sh

isri-aist · Dec 7, 2024 · 0eab92b · 0eab92b
1 parent 084c412
commit 0eab92b
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 15 deletions.
diff --git a/robo_manip_baselines/act/scripts/iterate_rollout.sh b/robo_manip_baselines/act/scripts/iterate_rollout.sh
@@ -1,12 +1,12 @@
 #!/bin/bash
 
-[[ $# < 1 ]] && echo "$0 <CKPT_DIR> [<CKPT_NAME> <TASK_NAME> <SKIP>]" && exit 1
+[[ $# < 1 ]] && echo "$0 <CHECKPOINT> <TASK_NAME> [<SKIP>]" && exit 1
 
-CKPT_DIR=$1
-TASK_NAME=${2:-MujocoUR5eCable}
+CHECKPOINT=$1
+TASK_NAME=$2
 SKIP=${3:-3}
 
-echo "[act/iterate_rollout.sh] CKPT_DIR: ${CKPT_DIR}"
+echo "[act/iterate_rollout.sh] CHECKPOINT: ${CHECKPOINT}"
 echo "[act/iterate_rollout.sh] TASK_NAME: ${TASK_NAME}"
 echo "[act/iterate_rollout.sh] SKIP: ${SKIP}"
 
@@ -17,7 +17,7 @@ WORLD_IDX_LIST=(0 1 2 3 4 5)
 for WORLD_IDX in "${WORLD_IDX_LIST[@]}"; do
     echo "[act/iterate_rollout.sh] WORLD_IDX: ${WORLD_IDX}"
     python ${SCRIPT_DIR}/../bin/rollout/RolloutAct${TASK_NAME}.py \
-           --checkpoint ${CKPT_DIR} \
+           --checkpoint ${CHECKPOINT} \
            --skip ${SKIP} \
            --world_idx ${WORLD_IDX} \
            --win_xy_policy 0 700 ${FIRST_OPTION}

diff --git a/robo_manip_baselines/diffusion_policy/scripts/iterate_rollout.sh b/robo_manip_baselines/diffusion_policy/scripts/iterate_rollout.sh
@@ -1,12 +1,12 @@
 #!/bin/bash
 
-[[ $# < 1 ]] && echo "$0 <CKPT_PATH> [<TASK_NAME> <SKIP>]" && exit 1
+[[ $# < 1 ]] && echo "$0 <CHECKPOINT> <TASK_NAME> [<SKIP>]" && exit 1
 
-CKPT_PATH=$1
-TASK_NAME=${2:-MujocoUR5eCable}
+CHECKPOINT=$1
+TASK_NAME=$2
 SKIP=${3:-3}
 
-echo "[diffusion_policy/iterate_rollout.sh] CKPT_PATH: ${CKPT_PATH}"
+echo "[diffusion_policy/iterate_rollout.sh] CHECKPOINT: ${CHECKPOINT}"
 echo "[diffusion_policy/iterate_rollout.sh] TASK_NAME: ${TASK_NAME}"
 echo "[diffusion_policy/iterate_rollout.sh] SKIP: ${SKIP}"
 
@@ -17,7 +17,7 @@ WORLD_IDX_LIST=(0 1 2 3 4 5)
 for WORLD_IDX in "${WORLD_IDX_LIST[@]}"; do
     echo "[diffusion_policy/iterate_rollout.sh] WORLD_IDX: ${WORLD_IDX}"
     python ${SCRIPT_DIR}/../bin/rollout/RolloutDiffusionPolicy${TASK_NAME}.py \
-           --checkpoint ${CKPT_PATH} \
+           --checkpoint ${CHECKPOINT} \
            --skip ${SKIP} \
            --world_idx ${WORLD_IDX} \
            --win_xy_policy 0 700 ${FIRST_OPTION}

diff --git a/robo_manip_baselines/sarnn/scripts/iterate_rollout.sh b/robo_manip_baselines/sarnn/scripts/iterate_rollout.sh
@@ -1,13 +1,13 @@
 #!/bin/bash
 
-[[ $# < 1 ]] && echo "$0 <CKPT_DIR> [<CKPT_NAME> <TASK_NAME> <CROPPED_IMG_SIZE> <SKIP>]" && exit 1
+[[ $# < 1 ]] && echo "$0 <CHECKPOINT> <TASK_NAME> [<CROPPED_IMG_SIZE> <SKIP>]" && exit 1
 
-CKPT_DIR=$1
-TASK_NAME=${2:-MujocoUR5eCable}
+CHECKPOINT=$1
+TASK_NAME=$2
 CROPPED_IMG_SIZE=${3:-280}
 SKIP=${4:-6}
 
-echo "[sarnn/iterate_rollout.sh] CKPT_DIR: ${CKPT_DIR}"
+echo "[sarnn/iterate_rollout.sh] CHECKPOINT: ${CHECKPOINT}"
 echo "[sarnn/iterate_rollout.sh] TASK_NAME: ${TASK_NAME}"
 echo "[sarnn/iterate_rollout.sh] CROPPED_IMG_SIZE: ${CROPPED_IMG_SIZE}"
 echo "[sarnn/iterate_rollout.sh] SKIP: ${SKIP}"
@@ -19,7 +19,7 @@ WORLD_IDX_LIST=(0 1 2 3 4 5)
 for WORLD_IDX in "${WORLD_IDX_LIST[@]}"; do
     echo "[sarnn/iterate_rollout.sh] WORLD_IDX: ${WORLD_IDX}"
     python ${SCRIPT_DIR}/../bin/rollout/RolloutSarnn${TASK_NAME}.py \
-           --checkpoint ${CKPT_DIR} \
+           --checkpoint ${CHECKPOINT} \
            --cropped_img_size ${CROPPED_IMG_SIZE} \
            --skip ${SKIP} \
            --world_idx ${WORLD_IDX} \