update

qhduan · Feb 17, 2018 · 8e76dcf · 8e76dcf
1 parent 43756cb
commit 8e76dcf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chatbot_ad/train_discriminative.py b/chatbot_ad/train_discriminative.py
@@ -28,7 +28,7 @@ def test(bidirectional, cell_type, depth,
         open('chatbot.pkl', 'rb'))
 
     # 训练部分
-    n_epoch = 5
+    n_epoch = 1
     batch_size = 512
     steps = int(len(x_data) / batch_size) + 1