Skip to content

Commit 4581524

Browse files
authored
Merge pull request #11350 from sneaxiy/argmin_argmax
Fix identifier error of 'dshape' in 'Paddle/benchmark/fluid/models/vgg.py'
2 parents 831909c + c2a00f0 commit 4581524

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

benchmark/fluid/models/vgg.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,8 @@ def get_model(args):
8282
data_file, batch_size=args.batch_size))
8383
images, label = fluid.layers.read_file(data_file)
8484
else:
85-
images = fluid.layers.data(name='data', shape=dshape, dtype='float32')
85+
images = fluid.layers.data(
86+
name='data', shape=data_shape, dtype='float32')
8687
label = fluid.layers.data(name='label', shape=[1], dtype='int64')
8788

8889
# Train program

0 commit comments

Comments
 (0)