restore old code of seqToseq_net

yijiuzai · Apr 10, 2017 · 42e9d7c · 42e9d7c
1 parent 6747c59
commit 42e9d7c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/demo/seqToseq/seqToseq_net.py b/demo/seqToseq/seqToseq_net.py
@@ -81,10 +81,8 @@ def gru_encoder_decoder(data_conf,
     """
     for k, v in data_conf.iteritems():
         globals()[k] = v
-    #source_dict_dim = len(open(src_dict_path, "r").readlines())
-    #target_dict_dim = len(open(trg_dict_path, "r").readlines())
-    source_dict_dim = 1000
-    target_dict_dim = 2000
+    source_dict_dim = len(open(src_dict_path, "r").readlines())
+    target_dict_dim = len(open(trg_dict_path, "r").readlines())
     gen_trans_file = gen_result
 
     src_word_id = data_layer(name='source_language_word', size=source_dict_dim)

diff --git a/demo/seqToseq/translation/train.conf b/demo/seqToseq/translation/train.conf
@@ -19,8 +19,7 @@ sys.path.append("..")
 from seqToseq_net import *
 
 # whether this config is used for generating
-#is_generating = False
-is_generating = True
+is_generating = False
 
 ### Data Definiation
 data_dir  = "./data/pre-wmt14"