Trainer: How can I log model outputs besides loss?

In my case, my custom model return “loss_1”, “loss_2” and “loss”, and loss = loss_1 + loss_2. But HF trainer only logs “loss” when training. I want to track all of them. How can I achieve this most easily?