Thanks so much!!!
May I suggest to add this information to the Trainer documentation, either at the subclassing Trainer example at the beginning or at the compute_loss part.
Thanks so much!!!
May I suggest to add this information to the Trainer documentation, either at the subclassing Trainer example at the beginning or at the compute_loss part.