export AIRBERT_ROOT=$(pwd)
export PYTHONPATH=${PYTHONPATH}:${AIRBERT_ROOT}/build

name=R2R-RC-VLN-BERT/sample-aug-init.airbert

flag="--vlnbert vilbert 
      --test_only 0
      --train validlistener

      --submit
      --features places365

      --maxAction 15
      --batchSize 8
      --feedback sample
      --lr 1e-5
      --iters 300000
      --optim adamW
      --mlWeight 0.20
      --maxInput 80
      --angleFeatSize 128
      --featdropout 0.4
      --dropout 0.5"

CUDA_VISIBLE_DEVICES=$1 python r2r_src/train.py $flag --name $name