Views
No views yet
.pt format1
2#!/bin/bash
3
4CKPT_LIST=(
5 "path/to/checkpoints/step-009000-epoch-34-loss=0.0760.pt"
6)
7
8# Loop over the checkpoint list and GPUs
9for i in "${!CKPT_LIST[@]}"; do
10 GPU_ID=$((i % 8)) # Cycle through GPUs 0-7
11 CHECKPOINT="${CKPT_LIST[$i]}"
12
13 # Run the evaluation script for each checkpoint and GPU
14 CUDA_VISIBLE_DEVICES=$GPU_ID python deploy/libero/run_libero_eval.py \
15 --model_family instruct_vla \
16 --pretrained_checkpoint "$CHECKPOINT" \
17 --task_suite_name libero_object \
18 --local_log_dir Libero/release_ensemble \
19 --use_length -1 \
20 --center_crop True &
21
22 # --use_length == -1 : execute the ensembled action
23 # --use_length >= 1 : execute action_chunk[0:use_length]
24 # For this checkpoint, you should use action ensemble.
25
26 sleep 5
27done
28
29# Wait for all background jobs to finish
30wait
31