d-robotics-rdt/rdt170m-train/process_data_rdt.sh
2025-11-02 12:25:30 +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