feat: original-version
This commit is contained in:
parent
93e8b23316
commit
007e160b13
@ -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')
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user