Skip to content

Fix auxiliary tasks with flow objects and update GN2X default params

Jackson Barr requested to merge jabarr/salt:jabarr-main-patch-49336 into main

Update track auxiliary tasks to ignore inputs from neutral flow objects. Also increases the default number of parameters in the GN2X model to bring in line with regular GN2. This also reflects the number of parameters used for the current technical taggers we have released.

The BCEWithLogitsLoss doesn't have a nice ignore_index like the regular cross entropy loss so I have modified the vertexing task directly, let me know if the way I have done it makes sense

Edited by Jackson Barr

Merge request reports