fix fea_only for forwarding

Soar-Sir · Jul 23, 2019 · 5cbffab · 5cbffab
1 parent 2f07d78
commit 5cbffab
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/data_io.py b/data_io.py
@@ -270,6 +270,7 @@ def _get_max_seq_length_from_config_str(config_str):
             max_seq_length=_get_max_seq_length_from_config_str(config['batches']['max_seq_length_valid'])
         if to_do=='forward':
             max_seq_length=-1 # do to break forward sentences
+            fea_only=True
         fea_dict, lab_dict, arch_dict = dict_fea_lab_arch(config, fea_only)
         seq_model = is_sequential_dict(config, arch_dict)
         return to_do, max_seq_length, fea_dict, lab_dict, arch_dict, seq_model