From 4d83fe6e207f4db492361803fee93ff389c3f298 Mon Sep 17 00:00:00 2001 From: Yan Li Date: Fri, 17 Jun 2016 15:54:32 +0800 Subject: [PATCH] update lstm.py reshape using shape changing target_shape to shape --- example/warpctc/lstm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/warpctc/lstm.py b/example/warpctc/lstm.py index 97fda6b9c9d4..32ba2455e11d 100644 --- a/example/warpctc/lstm.py +++ b/example/warpctc/lstm.py @@ -72,7 +72,7 @@ def lstm_unroll(num_lstm_layer, seq_len, hidden_concat = mx.sym.Concat(*hidden_all, dim=0) pred = mx.sym.FullyConnected(data=hidden_concat, num_hidden=11) - label = mx.sym.Reshape(data=label, target_shape=(0,)) + label = mx.sym.Reshape(data=label, shape=(-1,)) label = mx.sym.Cast(data = label, dtype = 'int32') sm = mx.sym.WarpCTC(data=pred, label=label, label_length = num_label, input_length = seq_len) return sm