From ff6a6c73552ad9b78108af390c0df267bb21c958 Mon Sep 17 00:00:00 2001
From: Yibing Liu <liuyibing01@baidu.com>
Date: Thu, 31 May 2018 20:17:57 -0700
Subject: [PATCH] Fix the bug in test

---
 fluid/DeepASR/train.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/fluid/DeepASR/train.py b/fluid/DeepASR/train.py
index 0111161a..8373c0e0 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)
-- 
GitLab