A PR to add tracking for individual losses (previously only total loss was tracked).
Now returns a dict of losses from the compute_loss method and logs all of them. The loss with key="loss" is the one used for training, the remaining ones are just tracked.
A PR to add tracking for individual losses (previously only total loss was tracked).
Now returns a dict of losses from the
compute_lossmethod and logs all of them. The loss withkey="loss"is the one used for training, the remaining ones are just tracked.