Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
trainer_utils.rst 797 Bytes
Newer Older
Sylvain Gugger's avatar
Trainer callbacks (#7596)  
Sylvain Gugger committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21 
Utilities for Trainer
-----------------------------------------------------------------------------------------------------------------------

This page lists all the utility functions used by :class:`~transformers.Trainer`.

Most of those are only useful if you are studying the code of the Trainer in the library.

Utilities
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: transformers.EvalPrediction

.. autofunction:: transformers.set_seed

.. autofunction:: transformers.torch_distributed_zero_first


Callbacks internals
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: transformers.trainer_callback.CallbackHandler