File tree Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -77,6 +77,7 @@ python demo_imagenet.py \
77
77
--save_dir='./save_qat_mbv2/' \
78
78
--devices='gpu' \
79
79
--batch_size=64 \
80
+ --data_dir='../data/ILSVRC2012/' \
80
81
--config_path='./configs/CV/mbv2_qat_dis.yaml'
81
82
```
82
83
@@ -90,6 +91,7 @@ python demo_imagenet.py \
90
91
--save_dir='./save_qat_mbv2/' \
91
92
--devices='gpu' \
92
93
--batch_size=64 \
94
+ --data_dir='../data/ILSVRC2012/' \
93
95
--config_path='./configs/CV/mbv2_ptq_hpo.yaml'
94
96
```
95
97
@@ -119,5 +121,6 @@ python demo_imagenet.py \
119
121
--save_dir='./save_qat_mbv2/' \
120
122
--devices='gpu' \
121
123
--batch_size=64 \
124
+ --data_dir='../data/ILSVRC2012/' \
122
125
--config_path='./configs/CV/xxx.yaml'
123
126
```
Original file line number Diff line number Diff line change @@ -33,7 +33,7 @@ def reader_wrapper(reader):
33
33
def gen ():
34
34
for i , data in enumerate (reader ()):
35
35
imgs = np .float32 ([item [0 ] for item in data ])
36
- yield {"inputs " : imgs }
36
+ yield {"x " : imgs }
37
37
38
38
return gen
39
39
You can’t perform that action at this time.
0 commit comments