d-robotics-rdt/RDT-170M/process_data_rdt.sh
2025-10-26 16:23:28 +08:00

9 lines
188 B
Bash

#!/bin/bash
task_name=${1}
task_config=${2}
expert_data_num=${3}
gpu_id=${4}
export CUDA_VISIBLE_DEVICES=${gpu_id}
python scripts/process_data.py $task_name $task_config $expert_data_num