Skip to content

Maskformer

This MR adds an initial working (version) of the MaskFormer model. Currently, we:

  • write the vertex index to the tracks, in a similar manner to other vertexing models
  • write the leading (defined as prediction with highest predicted pT) vertex information (pt, mass, Lxy, deta, dphi) to the jet, for comparisons

For this we still need the following:

  • validation (I'll throw some plots below)
  • Move the model onnx file to a central location and update the config

Description

Review checklist:

  • CI Passing
  • Comments addressed
  • Source branch is up to date with target

Merge request reports

Loading