Skip to content

Commit d01db56

Browse files
authored
fix ACT quick start demo dataloader error (PaddlePaddle#1123)
1 parent 4d3e418 commit d01db56

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

demo/auto_compression/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ class ImageNetDataset(DatasetFolder):
5858
train_dataset = ImageNetDataset("./ILSVRC2012_data_demo/ILSVRC2012/train/")
5959
image = paddle.static.data(
6060
name='inputs', shape=[None] + [3, 224, 224], dtype='float32')
61-
train_loader = paddle.io.DataLoader(train_dataset, feed_list=[image], batch_size=32)
61+
train_loader = paddle.io.DataLoader(train_dataset, feed_list=[image], batch_size=32, return_list=False)
6262
# 开始自动压缩
6363
ac = AutoCompression(
6464
model_dir="./MobileNetV1_infer/",

0 commit comments

Comments
 (0)