Pass logits and labels to losses using kwargs

2 jobs for loss in 51 minutes and 35 seconds (queued for 3 seconds)
Status Job ID Name Coverage
  Build
passed #137230
docker
build_linux_27

00:51:35

72.0%
passed #137231
docker
build_linux_36

00:49:19

72.0%