feat: original-version

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

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_seen', 'val_unseen']
val_env_names = ['val_seen', 'val_unseen']
if args.submit:
val_env_names.append('test')

View File

@ -10,7 +10,7 @@ obj_ft_dim=768
ngpus=1
seed=0
name=${train_alg}-${features}
name=${train_alg}-${features}-advanced_adversarial
name=${name}-seed.${seed} #-${ngpus}gpus
outdir=${DATA_ROOT}/REVERIE/exprs_map/finetune/${name}
@ -61,7 +61,7 @@ flag="--root_dir ${DATA_ROOT}
# train
CUDA_VISIBLE_DEVICES='0' python reverie/main_nav_obj.py $flag \
--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
# test