train.sh 259 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
4
5
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