From 799a3aa6bc27c1ceaa8e610f66ab51767cb8895c Mon Sep 17 00:00:00 2001 From: "yi.wu" Date: Mon, 11 Jun 2018 15:50:58 +0800 Subject: [PATCH] fix fluid_benchmark stacked_dynamic_lstm model error --- benchmark/fluid/models/stacked_dynamic_lstm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/benchmark/fluid/models/stacked_dynamic_lstm.py b/benchmark/fluid/models/stacked_dynamic_lstm.py index e1c4857f1a..211869af4e 100644 --- a/benchmark/fluid/models/stacked_dynamic_lstm.py +++ b/benchmark/fluid/models/stacked_dynamic_lstm.py @@ -104,8 +104,9 @@ def get_model(args): loss = fluid.layers.mean(x=loss) # add acc + batch_size_tensor = fluid.layers.create_tensor(dtype='int64') batch_acc = fluid.layers.accuracy(input=logit, label=fluid.layers.data(name='label', \ - shape=[1], dtype='int64')) + shape=[1], dtype='int64'), total=batch_size_tensor) inference_program = fluid.default_main_program().clone() with fluid.program_guard(inference_program): -- GitLab