name=visual
flag="--attn soft --train visual
      --featdropout 0.3
      --angleFeatSize 128
      --feedback sample
      --feature_size 2048
      --mlWeight 0.2
      --subout max --dropout 0.5 --optim rms --lr 1e-4
      --truncate last
      --con_weight 0.2
      --object_constraints
      --random_objects
      --sim ce
      --visual_next 0
      --visual_diff 0
      --iters 20000
      --batchSize 32
      --maxInput 160 
      --maxAction 7000"
mkdir -p snap/$name
CUDA_VISIBLE_DEVICES=$1 python r2r_src/train.py $flag --name $name
