  init. Note, that you can also subclass or override the create_optimizer and create_scheduler methods
  separately.
create_optimizer -- Sets up the optimizer if it wasn't passed at init.
create_scheduler -- Sets up the learning rate scheduler if it wasn't passed at init.
compute_loss - Computes the loss on a batch of training inputs.
training_step -- Performs a training step.
prediction_step -- Performs an evaluation/test step.
evaluate -- Runs an evaluation loop and returns metrics.
predict -- Returns predictions (with metrics if labels are available) on a test set.
The [Trainer] class is optimized for 🤗 Transformers models and can have surprising behaviors
when you use it on other models. When using it on your own model, make sure:
your model always return tuples or subclasses of [~file_utils.ModelOutput].
