diff --git a/fluid/DeepASR/train.py b/fluid/DeepASR/train.py index 0111161a235715d7969e732554e3cb41fadb4c9a..8373c0e04f3a4a2ae87d129243b519c0e0622144 100644 --- a/fluid/DeepASR/train.py +++ b/fluid/DeepASR/train.py @@ -195,6 +195,8 @@ def train(args): args.minimum_batch_size)): # load_data (features, labels, lod, _) = batch_data + features = np.reshape(features, (-1, 11, 3, args.frame_dim)) + features = np.transpose(features, (0, 2, 1, 3)) feature_t.set(features, place) feature_t.set_lod([lod]) label_t.set(labels, place)