8def AthExTritonCfg(flags, name="AthExTritonExample", **kwargs):
9 acc = ComponentAccumulator()
10
11 from AthTritonComps.TritonToolConfig import TritonToolCfg
12 kwargs.setdefault("InferenceTool", acc.popToolsAndMerge(
13 TritonToolCfg(flags, "MNIST_testModel", "localhost", name="EvaluateModelTritonTool")
14 ))
15
16
17 input_data = "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte"
18 kwargs.setdefault("BatchSize", 2)
19 kwargs.setdefault("InputDataPixel", input_data)
20 kwargs.setdefault("OutputLevel", Constants.DEBUG)
21 acc.addEventAlgo(CompFactory.AthInfer.ExampleMLInferenceWithTriton(name, **kwargs))
22
23 return acc
24