Pass logits and labels to losses using kwargs

Merge request reports

Loading