We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 831909c + c2a00f0 commit 4581524Copy full SHA for 4581524
benchmark/fluid/models/vgg.py
@@ -82,7 +82,8 @@ def get_model(args):
82
data_file, batch_size=args.batch_size))
83
images, label = fluid.layers.read_file(data_file)
84
else:
85
- images = fluid.layers.data(name='data', shape=dshape, dtype='float32')
+ images = fluid.layers.data(
86
+ name='data', shape=data_shape, dtype='float32')
87
label = fluid.layers.data(name='label', shape=[1], dtype='int64')
88
89
# Train program
0 commit comments