training cws on large corpus

cold11 · Jan 11, 2020 · 523e9d6 · 523e9d6
1 parent c61c0db
commit 523e9d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/train/zh/cws/train_large_rnn_cws.py b/tests/train/zh/cws/train_large_rnn_cws.py
@@ -11,7 +11,7 @@
 cdroot()
 
 tokenizer = RNNTokenizer()
-save_dir = 'data/model/cws/pku_6m_rnn_cws'
+save_dir = 'data/model/cws/large_rnn_cws'
 optimizer = tf.keras.optimizers.Adam(learning_rate=0.001,
                                      epsilon=1e-8, clipnorm=5)
 tokenizer.fit('data/cws/large/all.txt',