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