export PYTHONPATH=./
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
OUTPUT_DIR=outputs/vita_video_audio
bash script/train/finetuneTask_nodes.sh ${OUTPUT_DIR} # origin
# bash script/train/finetuneTask_nodes_singlenode.sh ${OUTPUT_DIR} # for try train