Skip to content

Temporary onnx export fix for GN3V01 with GLOBAL flag on

Wei Sheng Lairequested to merge
wlai/salt:wei/fix_global into main

Description

The current to_onnx script doesn't work for models with GLOBAL flag turn on. This MR fix that temporarily for GN3V01 where the only the jet features are copied and concatenated to the global representation after the transformer layer.

A pipeline test for GN3V01 is also added and aim to fix some naming conventions between flow and flows.

Relates to &7

Review checklist:

  • CI Passing
  • Comments addressed
  • Source branch is up to date with target
Edited by Wei Sheng Lai

Merge request reports