Typo fix in train script

harpribot · May 5, 2016 · 181c403 · 181c403
1 parent b240ddf
commit 181c403
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train_script_lstm_simple_no_attn.py b/train_script_lstm_simple_no_attn.py
@@ -12,6 +12,7 @@
 checkpointer.set_result_location(out_file)
 lstm_net = lstm_simple.NeuralNet(review_summary_file, checkpointer)
 lstm_net.set_parameters(train_batch_size=5,test_batch_size=25, memory_dim=50,learning_rate=0.05)
+lstm_net.begin_session()
 lstm_net.form_model_graph()
 lstm_net.fit()
 lstm_net.predict()