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)