From 875e07a9e3e8aac3f01b607d6c4f58678db8188a Mon Sep 17 00:00:00 2001 From: "Wesley A. Cheng" <15952538+wesleyacheng@users.noreply.github.com> Date: Tue, 29 Mar 2022 09:57:43 -0700 Subject: [PATCH] [doc] Fix missing trainer import (#16469) --- docs/source/training.mdx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/training.mdx b/docs/source/training.mdx index e6c4b146fa..0d3648a925 100644 --- a/docs/source/training.mdx +++ b/docs/source/training.mdx @@ -121,7 +121,7 @@ Call `compute` on `metric` to calculate the accuracy of your predictions. Before If you'd like to monitor your evaluation metrics during fine-tuning, specify the `evaluation_strategy` parameter in your training arguments to report the evaluation metric at the end of each epoch: ```py ->>> from transformers import TrainingArguments +>>> from transformers import TrainingArguments, Trainer >>> training_args = TrainingArguments(output_dir="test_trainer", evaluation_strategy="epoch") ```