[PWGEM,Tools] Update EMCal Conversion ML Model and enable batching of… - #17768
Conversation
… ml models - Add option to use ml models in batches, instead of testing one candidate per call to reduce speed up application of ml models - Update `emcalPhotonMcTask.cxx` to incoporate these changes - Update histograms that will be stored and filled
|
O2 linter results: ❌ 0 errors, |
|
Hello @fcatalan92 @fmazzasc |
|
Hi @mhemmer-cern , the PR looks good to me and I am approving it for the ML side. @ChSonnabend , could you maybe check the memory issue in model.h? It seems to me that @mhemmer-cern is right |
|
Hi all, |
alibuild
left a comment
There was a problem hiding this comment.
Auto-approving on behalf of @mhemmer-cern.
|
Hi @mhemmer-cern, thanks a lot for the developments and for spotting this issue! @ChSonnabend I guess that returning the outputValues by copy should be enough. OnnxRuntime also implements the possibility to return the value in a user-provided output https://github.com/microsoft/onnxruntime/blob/bb331b7a235435863a89f4bfcc3b2138928026ec/include/onnxruntime/core/session/onnxruntime_cxx_api.h#L1961 . |
… ml models
emcalPhotonMcTask.cxxto incoporate these changes