File tree 1 file changed +4
-4
lines changed
src/deepsparse/evaluation
1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 23
23
MODEL_PATH
24
24
A path to an ONNX model, local directory containing ONNX model
25
25
(including all the auxiliary files) or a SparseZoo stub
26
- -d DATASET , --dataset DATASET
26
+ -d DATASETS , --datasets DATASETS
27
27
The dataset to evaluate on. The user may pass multiple datasets
28
28
by passing the option multiple times.
29
29
-i INTEGRATION, --integration INTEGRATION
95
95
)
96
96
@click .option (
97
97
"-d" ,
98
- "--dataset " ,
98
+ "--datasets " ,
99
99
type = str ,
100
100
multiple = True ,
101
101
help = "The name of dataset to evaluate on. The user may pass multiple "
163
163
@click .argument ("integration_args" , nargs = - 1 , type = click .UNPROCESSED )
164
164
def main (
165
165
model_path ,
166
- dataset ,
166
+ datasets ,
167
167
integration ,
168
168
engine_type ,
169
169
save_path ,
@@ -182,7 +182,7 @@ def main(
182
182
183
183
"""
184
184
# join datasets to a list if multiple datasets are passed
185
- datasets = list (dataset ) if not isinstance (dataset , str ) else dataset
185
+ datasets = list (datasets ) if not isinstance (datasets , str ) else datasets
186
186
# format kwargs to a dict
187
187
integration_args = parse_kwarg_tuples (integration_args )
188
188
You can’t perform that action at this time.
0 commit comments