From 007e160b13df9987412821c97f6a47f2871a66c4 Mon Sep 17 00:00:00 2001
From: Ting-Jun Wang <levi900227@gmail.com>
Date: Sun, 28 Jan 2024 19:24:54 +0800
Subject: [PATCH] feat: original-version

---
 map_nav_src/reverie/main_nav_obj.py | 2 +-
 map_nav_src/scripts/run_reverie.sh  | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/map_nav_src/reverie/main_nav_obj.py b/map_nav_src/reverie/main_nav_obj.py
index 62721be..7c73a7d 100644
--- a/map_nav_src/reverie/main_nav_obj.py
+++ b/map_nav_src/reverie/main_nav_obj.py
@@ -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')
diff --git a/map_nav_src/scripts/run_reverie.sh b/map_nav_src/scripts/run_reverie.sh
index 69edfb7..fd72b69 100644
--- a/map_nav_src/scripts/run_reverie.sh
+++ b/map_nav_src/scripts/run_reverie.sh
@@ -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,11 +61,11 @@ 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
 CUDA_VISIBLE_DEVICES='0' python reverie/main_nav_obj.py $flag  \
       --tokenizer bert \
       --resume_file ../datasets/REVERIE/trained_models/best_val_unseen \
-      --test --submit
\ No newline at end of file
+      --test --submit