mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-05 22:00:09 +06:00
a1d1b332d0
* Add eval_accumulation_step and clean distributed eval * Add TPU test * Add TPU stuff * Fix arg name * Fix Seq2SeqTrainer * Fix total_size * Update src/transformers/trainer_pt_utils.py Co-authored-by: Lysandre Debut <lysandre@huggingface.co> * Doc and add test to TPU * Add unit test * Adapt name Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
27 lines
1.0 KiB
ReStructuredText
27 lines
1.0 KiB
ReStructuredText
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
|
|
|
|
Distributed Evaluation
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
.. autoclass:: transformers.trainer_pt_utils.DistributedTensorGatherer
|
|
:members: |