ONNX conversion - selvaggi/mlpf GitHub Wiki

To use PyTorch models in C++, they must first be converted to the ONNX format.

A conversion script is available at: scripts/onnx_conversion.py

The required environment is provided in the following Singularity image: /afs/cern.ch/work/l/lherrman/public/onnx_env.sif

To enter the environment, run:

apptainer shell onnx_env.sif

The Gatr model which is used for the conversion is not the official one but https://github.com/herrmannlena/gatr_onnx

For the conversion, the basis is needed which is described by the files geometric_product.pt and outer_product.pt They are accessible at/afs/cern.ch/work/l/lherrman/public/foronnx/