Accessing logits when using Trainer API

Hi,

This thread may help you: Metrics for Training Set in Trainer - #7 by Kaveri