snsd0805/adversarial_VLNBERT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: add val_train_seen

Browse Source
This commit is contained in:
Ting-Jun Wang 2023-11-19 13:35:09 +08:00
parent 9e5d2f95ba
commit dfe586b9ab
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
r2r_src/train.py
View File

@ -219,7 +219,7 @@ def train_val(test_only=False):
else: else:
featurized_scans = set([key.split("_")[0] for key in list(feat_dict.keys())]) featurized_scans = set([key.split("_")[0] for key in list(feat_dict.keys())])
# val_env_names = ['val_seen', 'val_unseen'] # 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'] # val_env_names = ['val_unseen']
train_env = R2RBatch(feat_dict, batch_size=args.batchSize, splits=['train'], tokenizer=tok) train_env = R2RBatch(feat_dict, batch_size=args.batchSize, splits=['train'], tokenizer=tok)