未验证 提交 45815247 编写于 作者: S sneaxiy 提交者: GitHub

Merge pull request #11350 from sneaxiy/argmin_argmax

Fix identifier error of 'dshape' in 'Paddle/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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册