#!/bin/bash/

export HSA_FORCE_FINE_GRAIN_PCIE=1
export USE_MIOPEN_BATCHNORM=1

python -m torch.distributed.launch --nproc_per_node=4 run_beit3_finetuning.py \
        --model beit3_base_patch16_480 \
        --input_size 480 \
        --task coco_captioning \
        --batch_size 16 \
        --sentencepiece_model ./pretrained_models/beit3.spm \
        --finetune ./pretrained_models/beit3_base_patch16_480_coco_captioning.pth \
        --data_path ../../data/coco2014/ \
        --output_dir ./save_models \
        --eval \
        --dist_eval