File size: 198 Bytes
ca19ab4
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
CUDA_VISIBLE_DEVICES=0,1,2,3 \
torchrun \
--nproc_per_node=4 \
--master_port 30010 \
train_caption.py \
--evaluate \
--eval_split test \
--config configs/caption_coco.yaml \
--output_dir output/blip