reverie's pretrain path

This commit is contained in:
Ting-Jun Wang 2024-09-07 18:15:31 +08:00
parent bfcb1f49ea
commit 36dbfed7e1
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354

View File

@ -10,7 +10,7 @@ obj_ft_dim=768
ngpus=1
seed=0
name=${train_alg}-${features}-advanced-adversarial
name=${train_alg}-${features}-new-reverie-only-front-path
name=${name}-seed.${seed} #-${ngpus}gpus
outdir=${DATA_ROOT}/REVERIE/exprs_map/finetune/${name}
@ -59,10 +59,10 @@ flag="--root_dir ${DATA_ROOT}
--gamma 0."
# train
# CUDA_VISIBLE_DEVICES='0' python3 reverie/main_nav_obj.py $flag \
# --tokenizer bert \
# --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
CUDA_VISIBLE_DEVICES='0' python3 reverie/main_nav_obj.py $flag \
--tokenizer bert \
--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
echo /root/mount/Matterport3DSimulator/VLN-DUET/datasets/REVERIE/exprs_map/finetune/${name}/ckpts/best_val_unseen