30 lines
655 B
Bash
30 lines
655 B
Bash
export AIRBERT_ROOT=$(pwd)
|
|
export PYTHONPATH=${PYTHONPATH}:${AIRBERT_ROOT}/build
|
|
|
|
name=REVERIE-RC-VLN-BERT-original/train-init.airbert-ver2
|
|
|
|
flag="--vlnbert vilbert
|
|
|
|
--train listener
|
|
--test_only 0
|
|
|
|
--init_bert_file snap/vln-bert/r2rM_bnbMS_2capt.pth1.4.bin
|
|
|
|
--features places365
|
|
--maxAction 15
|
|
--maxInput 50
|
|
--batchSize 8
|
|
--feedback sample
|
|
--lr 1e-5
|
|
--iters 200000
|
|
--log_every 1000
|
|
--optim adamW
|
|
|
|
--mlWeight 0.20
|
|
--angleFeatSize 128
|
|
--featdropout 0.4
|
|
--dropout 0.5"
|
|
|
|
mkdir -p snap/$name
|
|
CUDA_VISIBLE_DEVICES=$1 python3 reverie_src/train.py $flag --name $name
|