Skip to content

Onnx Export with variable Jet Dimensions

Jackson Barr requested to merge jabarr/salt:onnx_updated into main

Updating the ONNX export code to allow the exporting of models with a varying number of input jet features (e.g. GN2X with pT, eta and mass)

Edited by Jackson Barr

Merge request reports