Skip to content

P
PaddleSlim

PaddlePaddle / PaddleSlim
接近 2 年 前同步成功

通知 51
Star 1434
Fork 344
P
PaddleSlim
未验证 提交 d01db560 编写于 作者: G Guanghua Yu 提交者： GitHub
浏览文件
操作

fix ACT quick start demo dataloader error (#1123)

上级 4d3e4184
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
demo/auto_compression/README.md
浏览文件 @ d01db560
...@@ -58,7 +58,7 @@ class ImageNetDataset(DatasetFolder): ...@@ -58,7 +58,7 @@ class ImageNetDataset(DatasetFolder):
train_dataset = ImageNetDataset("./ILSVRC2012_data_demo/ILSVRC2012/train/") train_dataset = ImageNetDataset("./ILSVRC2012_data_demo/ILSVRC2012/train/")
image = paddle.static.data( image = paddle.static.data(
name='inputs', shape=[None] + [3, 224, 224], dtype='float32') name='inputs', shape=[None] + [3, 224, 224], dtype='float32')
train_loader = paddle.io.DataLoader(train_dataset, feed_list=[image], batch_size=32) train_loader = paddle.io.DataLoader(train_dataset, feed_list=[image], batch_size=32, return_list=False)
# 开始自动压缩 # 开始自动压缩
ac = AutoCompression( ac = AutoCompression(
model_dir="./MobileNetV1_infer/", model_dir="./MobileNetV1_infer/",
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或