From dfe586b9ab028939d450a2e3e3787b8708cd9fd2 Mon Sep 17 00:00:00 2001 From: Ting-Jun Wang Date: Sun, 19 Nov 2023 13:35:09 +0800 Subject: [PATCH] fix: add val_train_seen --- r2r_src/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/r2r_src/train.py b/r2r_src/train.py index fc7e151..a7a9b28 100644 --- a/r2r_src/train.py +++ b/r2r_src/train.py @@ -219,7 +219,7 @@ def train_val(test_only=False): else: featurized_scans = set([key.split("_")[0] for key in list(feat_dict.keys())]) # val_env_names = ['val_seen', 'val_unseen'] - val_env_names = ['train', 'val_unseen'] + val_env_names = ['train', 'val_seen', 'val_unseen'] # val_env_names = ['val_unseen'] train_env = R2RBatch(feat_dict, batch_size=args.batchSize, splits=['train'], tokenizer=tok)