diff --git a/fluid/DeepASR/tools/profile.py b/fluid/DeepASR/tools/profile.py index 69aee88e22d33ed80212692bf61e41e1666bf5e5..8d720c16cd0ec6a9d4bb533a878b07973ced7176 100644 --- a/fluid/DeepASR/tools/profile.py +++ b/fluid/DeepASR/tools/profile.py @@ -168,7 +168,7 @@ def profile(args): start_time = time.time() frames_seen = 0 # load_data - (features, labels, lod) = batch_data + (features, labels, lod, _) = batch_data feature_t.set(features, place) feature_t.set_lod([lod]) label_t.set(labels, place) diff --git a/fluid/DeepASR/train.py b/fluid/DeepASR/train.py index 3908a550cdcf095057ea6ab0b89e07dcecda51f9..be99998c8aa7f88d49dab711e94dcd7cfef042d6 100644 --- a/fluid/DeepASR/train.py +++ b/fluid/DeepASR/train.py @@ -192,7 +192,7 @@ def train(args): test_data_reader.batch_iterator(args.batch_size, args.minimum_batch_size)): # load_data - (features, labels, lod) = batch_data + (features, labels, lod, _) = batch_data feature_t.set(features, place) feature_t.set_lod([lod]) label_t.set(labels, place)