fix: change the model path

This commit is contained in:
Ting-Jun Wang 2024-01-22 00:42:40 +08:00
parent 924cfe9b43
commit 0135ab3ac8
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354

View File

@ -59,13 +59,12 @@ flag="--root_dir ${DATA_ROOT}
--gamma 0."
# 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' \
--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
CUDA_VISIBLE_DEVICES='0' python reverie/main_nav_obj.py $flag \
CUDA_VISIBLE_DEVICES='0' python3 reverie/main_nav_obj.py $flag \
--tokenizer bert \
--resume_file ../datasets/REVERIE/trained_models/best_val_unseen \
--resume_file /root/mount/Matterport3DSimulator/VLN-DUET/datasets/REVERIE/exprs_map/finetune/dagger-vitbase-seed.0/ckpts/best_val_unseen \
--test --submit