name=reverie-v4.2.1 flag="--description reverie-v4.2.1 --train listener --features places365 --test_only 0 --maxAction 15 --maxInput 50 --batchSize 4 --feedback sample --lr 1e-5 --iters 200000 --optim adamW --attn soft --mlWeight 0.20 --featdropout 0.4 --angleFeatSize 128 --subout max --dropout 0.5" mkdir -p snap/$name CUDA_VISIBLE_DEVICES=0 python3 r2r_src/train.py $flag --name $name