10def AthExOnnxRuntimeExampleCfg(flags, name="AthOnnxExample", **kwargs):
11 acc = ComponentAccumulator()
12
13 model_fname = "dev/MLTest/2020-03-02/MNIST_testModel.onnx"
14 execution_provider = OnnxRuntimeType.CPU
15 kwargs.setdefault("ORTInferenceTool", acc.popToolsAndMerge(
16 OnnxRuntimeInferenceToolCfg(flags, model_fname, execution_provider)
17 ))
18
19 input_data = "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte"
20 kwargs.setdefault("BatchSize", 100)
21 kwargs.setdefault("InputDataPixel", input_data)
22 kwargs.setdefault("OutputLevel", Constants.DEBUG)
23 acc.addEventAlgo(
24 CompFactory.AthOnnx.EvaluateModelWithAthInfer(name, **kwargs))
25
26 return acc
27
28