From 4b6d584f4fb3741fe3d3268c36b54b8469444f60 Mon Sep 17 00:00:00 2001 From: sneaxiy Date: Mon, 11 Jun 2018 06:01:41 +0000 Subject: [PATCH] fix identifier error of 'dshape' --- benchmark/fluid/models/vgg.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/benchmark/fluid/models/vgg.py b/benchmark/fluid/models/vgg.py index 6092cdeb88..932601302d 100644 --- a/benchmark/fluid/models/vgg.py +++ b/benchmark/fluid/models/vgg.py @@ -82,7 +82,8 @@ def get_model(args): data_file, batch_size=args.batch_size)) images, label = fluid.layers.read_file(data_file) else: - images = fluid.layers.data(name='data', shape=dshape, dtype='float32') + images = fluid.layers.data( + name='data', shape=data_shape, dtype='float32') label = fluid.layers.data(name='label', shape=[1], dtype='int64') # Train program -- GitLab