Transformers 4.37 中文文档(十九)

2024-06-26 15:18:55 浏览数 (1)

原文: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 类针对

0 人点赞