Spaces:

yuezih
/

BLIP-SMILE

Runtime error

App Files Files Community

BLIP-SMILE / SMILE /BLIP /scripts /train.sh

yuezih

init

ca19ab4 over 1 year ago

165 Bytes

	CUDA_VISIBLE_DEVICES=1,2,3,4 \
	torchrun \
	--nproc_per_node=4 \
	--master_port 30000 \
	train_caption.py \
	--config configs/caption_coco.yaml \
	--output_dir output/blip