How to log a new tensor variable in TrainingArguments
I am using Seq2SeqTrainer class for my generation task. And I am using 2 losses in the Decoder class. I am using loss1, loss2, and loss=loss1+loss2. Currently only loss is getting logged as per log_steps argument. But I do want to log other variable such as loss1 and loss2. How to achieve that ?