chanhee-luke / m-track Goto Github PK

bash run/train_agent.bash
flag="--vlnbert prevalent_alfred
--test_only 0
--train listener
--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
--split train
"

mkdir -p snap/$name
CUDA_VISIBLE_DEVICES=3 python r2r_src/train.py $flag --name $name $@

你压根没有r2r_src这个文件夹，只有mtrack_src这个文件夹，而且压根跑不起来