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?
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?