ATLAS Offline Software
Loading...
Searching...
No Matches
AthExTriton_test Namespace Reference

Functions

 AthExTritonCfg (flags, name="AthExTritonExample", **kwargs)

Variables

 flags = initConfigFlags()
 acc = MainServicesCfg(flags)
 withDetails
 True
 summariseProps

Function Documentation

◆ AthExTritonCfg()

AthExTriton_test.AthExTritonCfg ( flags,
name = "AthExTritonExample",
** kwargs )

Definition at line 8 of file AthExTriton_test.py.

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

Variable Documentation

◆ acc

AthExTriton_test.acc = MainServicesCfg(flags)

Definition at line 35 of file AthExTriton_test.py.

◆ flags

AthExTriton_test.flags = initConfigFlags()

Definition at line 32 of file AthExTriton_test.py.

◆ summariseProps

AthExTriton_test.summariseProps

Definition at line 37 of file AthExTriton_test.py.

◆ True

AthExTriton_test.True

Definition at line 37 of file AthExTriton_test.py.

◆ withDetails

AthExTriton_test.withDetails

Definition at line 37 of file AthExTriton_test.py.