Skip to content
Snippets Groups Projects

Adding provision for batch execution for onnx models and renaming some old files

Merged Debottam Bakshi Gupta requested to merge dbakshig/athena:onnxExtra into master
6 files
+ 358
40
Compare changes
  • Side-by-side
  • Inline
Files
6
@@ -5,17 +5,21 @@ from AthenaCommon.AlgSequence import AlgSequence
algSequence = AlgSequence()
# Set up the job.
from AthExOnnxRuntime.AthExOnnxRuntimeConf import AthONNX__CxxApiAlgorithm
from AthExOnnxRuntime.AthExOnnxRuntimeConf import AthONNX__EvaluateModel
from AthOnnxruntimeService.AthOnnxruntimeServiceConf import AthONNX__ONNXRuntimeSvc
from AthenaCommon.AppMgr import ServiceMgr
ServiceMgr += AthONNX__ONNXRuntimeSvc( OutputLevel = DEBUG )
algSequence += AthONNX__CxxApiAlgorithm("AthONNX")
algSequence += AthONNX__EvaluateModel("AthONNX")
# Get a random no. between 0 to 10k for test sample
from random import randint
AthONNX = algSequence.AthONNX
AthONNX.TestSample = randint(0, 9999)
AthONNX.DoBatches = False
AthONNX.NumberOfBatches = 1
AthONNX.SizeOfBatch = 1
# Run for 10 "events".
theApp.EvtMax = 10
theApp.EvtMax = 2
Loading