Update babi_memnn.py

mgq1507 · Mar 16, 2017 · 1a0792a · 1a0792a
1 parent b5a0239
commit 1a0792a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/babi_memnn.py b/examples/babi_memnn.py
@@ -214,5 +214,5 @@ def vectorize_stories(data, word_idx, story_maxlen, query_maxlen):
 # train
 model.fit([inputs_train, queries_train], answers_train,
            batch_size=32,
-           epochs=3,
+           epochs=120,
            validation_data=([inputs_test, queries_test], answers_test))