From 4a2decd19280474fdaccd96312e8bda09cd85228 Mon Sep 17 00:00:00 2001 From: guspuffygit <85897841+guspuffygit@users.noreply.github.com> Date: Wed, 14 May 2025 08:40:00 -0400 Subject: [PATCH] Update trainer.md (#38113) Fix typo in torch.compile method parameters --- docs/source/en/trainer.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/source/en/trainer.md b/docs/source/en/trainer.md index 892594a1bc..b83d9ed60b 100644 --- a/docs/source/en/trainer.md +++ b/docs/source/en/trainer.md @@ -372,14 +372,14 @@ accelerate launch \ ### torch.compile -[torch.compile](./perf_torch_compile) can significantly speed up training and reduce computational overhead. Configure your torch.compile settings in [`TrainingArguments`]. Set `torch.compile` to `True`, and select a backend and compile mode. +[torch.compile](./perf_torch_compile) can significantly speed up training and reduce computational overhead. Configure your torch.compile settings in [`TrainingArguments`]. Set `torch_compile` to `True`, and select a backend and compile mode. ```py from transformers import TrainingArguments training_args = TrainingArguments( - torch.compile=True, - torch.compile_backend="inductor", + torch_compile=True, + torch_compile_backend="inductor", torch_compile_mode="default", ..., )