[go: up one dir, main page]
Skip to content

BCE loss for jet classification

Sophie Gelinasrequested to merge
sgelinas/salt:bce_loss into main

Description

Adds the option to use BCEWithLogitsLoss instead of CrossEntropyLoss as the loss function for the jet classification task.

Review checklist:

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

Merge request reports