9def AthExOnnxRuntimeExampleCfg(flags, name="AthOnnxExample", **kwargs):
10 acc = ComponentAccumulator()
11
12 model_fname = "dev/MLTest/2020-03-02/MNIST_testModel.onnx"
13 execution_provider = OnnxRuntimeType.CPU
14 from AthOnnxComps.OnnxRuntimeInferenceConfig import OnnxRuntimeInferenceToolCfg
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", 3)
21 kwargs.setdefault("InputDataPixel", input_data)
22 kwargs.setdefault("OutputLevel", Constants.DEBUG)
23 acc.addEventAlgo(CompFactory.AthOnnx.EvaluateModel(name, **kwargs))
24
25 return acc
26