snsd0805/adversarial_VLNDUET
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: original-version

Browse Source
This commit is contained in:
Ting-Jun Wang 2024-01-28 19:24:54 +08:00
parent 93e8b23316
commit 007e160b13
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map_nav_src/reverie/main_nav_obj.py
View File

@ -66,7 +66,7 @@ def build_dataset(args, rank=0):
) )
# val_env_names = ['val_train_seen'] # val_env_names = ['val_train_seen']
val_env_names = ['val_train_seen', 'val_seen', 'val_unseen'] val_env_names = ['val_seen', 'val_unseen']
if args.submit: if args.submit:
val_env_names.append('test') val_env_names.append('test')

6
map_nav_src/scripts/run_reverie.sh
View File

@ -10,7 +10,7 @@ obj_ft_dim=768
ngpus=1 ngpus=1
seed=0 seed=0
name=${train_alg}-${features} name=${train_alg}-${features}-advanced_adversarial
name=${name}-seed.${seed} #-${ngpus}gpus name=${name}-seed.${seed} #-${ngpus}gpus
outdir=${DATA_ROOT}/REVERIE/exprs_map/finetune/${name} outdir=${DATA_ROOT}/REVERIE/exprs_map/finetune/${name}
@ -61,11 +61,11 @@ flag="--root_dir ${DATA_ROOT}
# train # train
CUDA_VISIBLE_DEVICES='0' python reverie/main_nav_obj.py $flag \ CUDA_VISIBLE_DEVICES='0' python reverie/main_nav_obj.py $flag \
--tokenizer bert \ --tokenizer bert \
--bert_ckpt_file 'put the pretrained model (see pretrain_src) here' \ --bert_ckpt_file '../datasets/REVERIE/exprs_map/pretrain/cmt-vitbase-mlm.mrc.sap.og-init.lxmert-aug.speaker/ckpts/model_step_100000.pt' \
--eval_first --eval_first
# test # test
CUDA_VISIBLE_DEVICES='0' python reverie/main_nav_obj.py $flag \ CUDA_VISIBLE_DEVICES='0' python reverie/main_nav_obj.py $flag \
--tokenizer bert \ --tokenizer bert \
--resume_file ../datasets/REVERIE/trained_models/best_val_unseen \ --resume_file ../datasets/REVERIE/trained_models/best_val_unseen \
--test --submit --test --submit