ray.train.huggingface.transformers.prepare_trainer

ray.train.huggingface.transformers.prepare_trainer(trainer: transformers.Trainer) → transformers.Trainer

为 Ray Train 准备您的 HuggingFace Transformer Trainer。

此实用函数使训练器能够与 Ray 数据集成集成。在内部，它会覆盖 get_train_dataloader 和 get_eval_dataloader 方法，并在 train_dataset 和 eval_dataset 是 Ray 数据可迭代对象时注入数据集成逻辑。

PublicAPI (测试版): 此API目前处于测试阶段，在成为稳定版本之前可能会发生变化。