Fix auxiliary tasks with flow objects and update GN2X default params
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