How to use the multiple output of the model while calling Trainer
The outputs of my model consisted of several elements including loss, logits, new_tensor. While calling Trainer API, the predictions and label_ids will be passed into pred. I was wondering how to get the other outputs such as logits and new_tensor?