train_vqa.sh (2086B)
1 #!/bin/bash 2 # 3 # 4 # Comments starting with #OAR are used by the resource manager if using "oarsub -S" 5 # 6 # Note : quoting style of parameters matters, follow the example 7 # 8 # The job is submitted to the default queue 9 #OAR -q default 10 # Path to the binary to run 11 12 module purge 13 module load conda 14 source activate pytorch 15 module load cuda/10.2 16 module load gcc/7.3.0 17 module load cudnn/7.6-cuda-10.2 18 19 model=$1 20 unfreeze=$2 21 learningrate=$3 22 subsamble_other=$4 23 24 echo python -m mmsrl.train \ 25 --ofa_filename=$model \ 26 configs/ofa_vqa.py --unfreeze=$unfreeze --learning_rate=$learningrate --subsamble_other=$subsamble_other\ 27 --output_val=\"outputs/ofa_vqa_${model}-${unfreeze}-${learningrate}-${subsamble_other}_val.pkl\"\ 28 --output_test=\"outputs/ofa_vqa_${model}-${unfreeze}-${learningrate}-${subsamble_other}_test.pkl\"\ 29 --save_model=\"models/ofa_vqa_${model}-${unfreeze}-${learningrate}-${subsamble_other}.bin\" 30 31 python -m mmsrl.train --ofa_filename=$model configs/ofa_vqa.py --ofa_path=ofa_dir+ofa_filename --unfreeze=$unfreeze --learning_rate=$learningrate --subsamble_other=$subsamble_other\ 32 --output_val=\"outputs/ofa_vqa_${model}-${unfreeze}-${learningrate}-${subsamble_other}_val.pkl\"\ 33 --output_test=\"outputs/ofa_vqa_${model}-${unfreeze}-${learningrate}-${subsamble_other}_test.pkl\"\ 34 --save_model=\"models/ofa_vqa_${model}-${unfreeze}-${learningrate}-${subsamble_other}.bin\" 35 36 # oarsub -p "gpu='YES'" -l /gpunum=1,walltime=20:00:00 -t besteffort ./oarscripts/train_vqa.sh 37 # oarsub -p "gpu='YES' and host='nefgpu39.inria.fr'" -l /gpunum=1,walltime=20:00:00 -t besteffort ./oarscripts/train_vqa.sh 38 # oarsub -p "gpu='YES' and host='nefgpu50.inria.fr'" -l /nodes=1/gpunum=1,walltime=20:00:00 --array-param-file oarscripts/ofa_params.txt -t besteffort ./oarscripts/train_vqa.sh 39 40 41 #OAR_JOB_ID=12236582 42 # "--ofa_input='ocr question'" \