"tests/mbart/test_modeling_tf_mbart.py" did not exist on "e73a3e1891775a915846cc0f24b7e9a26d6688fb"
val.sh 482 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
5
6
7
8
#!/bin/bash/

export HSA_FORCE_FINE_GRAIN_PCIE=1
export USE_MIOPEN_BATCHNORM=1

python run.py with data_root=/public/home/chenych/datasets/coco2014/ num_gpus=1 num_nodes=1 "task_finetune_irtr_coco_base_image384" per_gpu_batchsize=64 load_path="./pretrained_models/vlmo_base_patch16_384_coco.pt" test_only=True

# python run.py with data_root=<ARROW_ROOT> num_gpus=1 num_nodes=1 "<CONFIG_NAME>" per_gpu_batchsize=<BS_FITS_YOUR_GPU> load_path="<Finetuned_VLMo_WEIGHT>" test_only=True