diff --git a/example/rnn/lstm.py b/example/rnn/lstm.py index 33fdb507f0dd..b560d5d270cf 100644 --- 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]