wehub-resource-sync
e06fe8e8c6
Secret Leaks / trufflehog (push) Failing after 1s
Build documentation / build (push) Failing after 1s
Build documentation / build_other_lang (push) Failing after 0s
CodeQL Security Analysis / CodeQL Analysis (push) Failing after 0s
PR CI / pr-ci (push) Failing after 1s
Slow tests on important models (on Push - A10) / Get all modified files (push) Failing after 1s
Slow tests on important models (on Push - A10) / Model CI (push) Has been skipped
Self-hosted runner (benchmark) / Benchmark (aws-g5-4xlarge-cache) (push) Has been cancelled
New model PR merged notification / Notify new model (push) Has been cancelled
Update Transformers metadata / build_and_package (push) Has been cancelled
1.7 KiB
1.7 KiB
Trainer
[Trainer] is a complete training and evaluation loop for Transformers models. You only need a model and dataset to get started.
Underneath, [Trainer] handles batching, shuffling, and padding your dataset into tensors. The training loop runs the forward pass, calculates loss, backpropagates gradients, and updates weights. Configure the training run with [TrainingArguments] to customize everything from batch size and training duration to distributed strategies, compilation, and more.
Next steps
- Start with the fine-tuning tutorial for an introduction to training a large language model with [
Trainer]. - Check the Subclassing Trainer methods guide for examples of how to subclass [
Trainer] methods. - See the Data collators guide to learn how to create a data collator for custom batch assembly.
- See the Callbacks guide to learn how to hook into training events.