原文:
huggingface.co/docs/transformers
训练器
原始文本:
huggingface.co/docs/transformers/v4.37.2/en/main_classes/trainer
Trainer 类提供了一个用于在 PyTorch 中进行完整特征训练的 API,并支持在多个 GPU/TPU 上进行分布式训练,支持NVIDIA GPUs的混合精度,AMD GPUs,以及 PyTorch 的torch.amp
。Trainer 与 TrainingArguments 类相辅相成,后者提供了广泛的选项来自定义模型的训练方式。这两个类一起提供了一个完整的训练 API。
Seq2SeqTrainer 和 Seq2SeqTrainingArguments 继承自 Trainer 和TrainingArgument
类,它们适用于用于序列到序列任务(如摘要或翻译)的模型训练。
Trainer 类针对