Skip to content

ONNX changes for GN3

Nikita Ivvan Pond requested to merge npond/salt:onnx_combime_classes into main

Description

Implements a series of changes for exporting GN3.

  • Renaming of global tasks, e.g. pghostsplitbjets -> pb
  • Combining global tasks, e.g pu = ps + pud + pgluon
  • Dumps the 'track type' task
  • Modified the checking code to account for second global tasks (e.g. jet regression) and task type.

TODO

  • Update onnx and onnxruntime to latest version - I'm apprehensive to do this unless we need to, as some UCL HPCs don't currently support it, so it would mean we have to run things in containers.

Review checklist:

  • CI Passing
  • Comments addressed
  • Source branch is up to date with target
Edited by Nikita Ivvan Pond

Merge request reports