nas / LAVIS-main /run_scripts /blip2 /train /train_caption_coco.sh
yuccaaa's picture
Add files using upload-large-folder tool
9627ce0 verified
python -m torch.distributed.run --nproc_per_node=16 train.py --cfg-path lavis/projects/blip2/train/caption_coco_ft.yaml