do not evaluate on 'train_subset' during training

ronghanghu · Sep 17, 2018 · b4acf98 · b4acf98
1 parent 1506b29
commit b4acf98
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tasks/R2R/train.py b/tasks/R2R/train.py
@@ -211,7 +211,8 @@ def make_env_and_models(args, train_vocab_path, train_splits, test_splits,
 
 def train_setup(args, batch_size=BATCH_SIZE):
  train_splits = ['train']
- val_splits = ['train_subset', 'val_seen', 'val_unseen']
+ # val_splits = ['train_subset', 'val_seen', 'val_unseen']
+ val_splits = ['val_seen', 'val_unseen']
  if args.use_test_set:
  val_splits.append('test')
 

diff --git a/tasks/R2R/train_speaker.py b/tasks/R2R/train_speaker.py
@@ -205,7 +205,8 @@ def make_env_and_models(args, train_vocab_path, train_splits, test_splits,
 
 def train_setup(args):
  train_splits = ['train']
- val_splits = ['train_subset', 'val_seen', 'val_unseen']
+ # val_splits = ['train_subset', 'val_seen', 'val_unseen']
+ val_splits = ['val_seen', 'val_unseen']
  vocab = TRAIN_VOCAB
 
  if args.use_train_subset: