Merge pull request #45 from cmry/master

no val lexicalizer fix
UFAL-DSG · Jul 2, 2020 · b925037 · b925037
2 parents 6ccd915 + 3f92cfc
commit b925037
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tgen/seq2seq.py b/tgen/seq2seq.py
@@ -448,6 +448,8 @@ def _init_training(self, das_file, ttree_file, data_portion,
         # ... or save part of the training data for validation:
         elif self.validation_size > 0:
             valid_trees_for_lexic, valid_idxs = self._cut_valid_data()  # will set train_trees, valid_trees, train_das, valid_das
+        else:
+            valid_trees_for_lexic, valid_idxs = None, None
 
         if self.validation_use_all_refs:  # try to use multiple references (not in lexicalizer)
             self._regroup_valid_refs()