Produce ML prediction of virtual photon four momentum, implemented in ONNX - #278

Open

cpecar wants to merge 11 commits into

mainfrom

onnx-reco-predictions

Commits on May 1, 2023

Start of changes to produce NN predictions with ONNX
cpecar
committed
515da85

Commits on May 2, 2023

setting 0 padding for ML prediction
cpecar
committed
3581f16
script to download onnx .tar file
cpecar
committed
2c9561f
updated makefile for onnx changes (need to make onnx build optional)
cpecar
committed
b8b96ae
added onnx includes to environment setup script (need to make optional)
cpecar
committed
67aca34
Fixed errors in setting tensor dimension in onnx. Hard-coded input tensor names (need to check if onnx has some better workaround, or if I can fix these for all models)
cpecar
committed
aa63f2b

Commits on May 3, 2023

updated testml macro to reflect onnx changes
cpecar
committed
c7106ae
started script to install onnx
cpecar
committed
932e1fd
Now pass onnx file name through macro, rather than hard codin
cpecar
committed
477d424

Commits on May 4, 2023

Make including onnx-runtime optional
cpecar
committed
c637973

Commits on May 5, 2023

Merge branch 'main' into onnx-reco-predictions
c-dilks
authored
9f6f39b