Merge pull request apache#1006 from qiaohaijun/patch-8

correct train's output
yzhliu · Dec 21, 2015 · d417d9a · d417d9a
2 parents 19863ea + afddc97
commit d417d9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/example/rnn/lstm.py b/example/rnn/lstm.py
@@ -187,8 +187,8 @@ def calc_nll(seq_label_probs, X, begin):
 def train_lstm(model, X_train_batch, X_val_batch,
                num_round, update_period,
                optimizer='rmsprop', half_life=2,max_grad_norm = 5.0, **kwargs):
-    print("Training swith train.shape=%s" % str(X_train_batch.shape))
-    print("Training swith val.shape=%s" % str(X_val_batch.shape))
+    print("Training with train.shape=%s" % str(X_train_batch.shape))
+    print("Training with val.shape=%s" % str(X_val_batch.shape))
     m = model
     seq_len = len(m.seq_data)
     batch_size = m.seq_data[0].shape[0]