8def AthExTritonCfg(flags, name="AthExTritonExample", **kwargs):
9 acc = ComponentAccumulator()
10
11 from AthTritonComps.TritonToolConfig import TritonToolCfg
12 try:
13 kwargs.setdefault("InferenceTool", acc.popToolsAndMerge(
14 TritonToolCfg(flags, "MNIST_testModel", "localhost",
15 name="EvaluateModelTritonTool")
16 ))
17 except RuntimeError as e:
18 import sys
19 from AthenaCommon.Logging import log as msg
20 msg.warning(e)
21 sys.exit(2)
22
23 input_data = "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte"
24 kwargs.setdefault("BatchSize", 2)
25 kwargs.setdefault("InputDataPixel", input_data)
26 kwargs.setdefault("OutputLevel", Constants.DEBUG)
27 acc.addEventAlgo(
28 CompFactory.AthInfer.ExampleAsyncMLInferenceWithTriton(name, **kwargs))
29
30 return acc
31
32