Skip to content

Commit bd4ecf0

Browse files
authored
fix ac imagenet demo (PaddlePaddle#1063)
* fix demo * fix demo
1 parent d3ecb04 commit bd4ecf0

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

demo/auto-compression/README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ python demo_imagenet.py \
7777
--save_dir='./save_qat_mbv2/' \
7878
--devices='gpu' \
7979
--batch_size=64 \
80+
--data_dir='../data/ILSVRC2012/' \
8081
--config_path='./configs/CV/mbv2_qat_dis.yaml'
8182
```
8283

@@ -90,6 +91,7 @@ python demo_imagenet.py \
9091
--save_dir='./save_qat_mbv2/' \
9192
--devices='gpu' \
9293
--batch_size=64 \
94+
--data_dir='../data/ILSVRC2012/' \
9395
--config_path='./configs/CV/mbv2_ptq_hpo.yaml'
9496
```
9597

@@ -119,5 +121,6 @@ python demo_imagenet.py \
119121
--save_dir='./save_qat_mbv2/' \
120122
--devices='gpu' \
121123
--batch_size=64 \
124+
--data_dir='../data/ILSVRC2012/' \
122125
--config_path='./configs/CV/xxx.yaml'
123126
```

demo/auto-compression/demo_imagenet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def reader_wrapper(reader):
3333
def gen():
3434
for i, data in enumerate(reader()):
3535
imgs = np.float32([item[0] for item in data])
36-
yield {"inputs": imgs}
36+
yield {"x": imgs}
3737

3838
return gen
3939

0 commit comments

Comments
 (0)