Pass logits and labels to losses using kwargs

Merged Amir MOHAMMADI requested to merge loss into master

Merge request reports