Latte-1 / train_scripts /ucf101_train.sh
maxin-cn's picture
Upload folder using huggingface_hub
94bafa8 verified
raw
history blame contribute delete
143 Bytes
export CUDA_VISIBLE_DEVICES=4,5
torchrun --nnodes=1 --nproc_per_node=2 --master_port=29509 train.py --config ./configs/ucf101/ucf101_train.yaml