Skip to content

Commit

Permalink
Update training.md (#716)
Browse files Browse the repository at this point in the history
Fix to TrainingArguments
TrainingsArguments doesn't exist in the transformers library
  • Loading branch information
Kotstantinovskiy authored Jul 10, 2024
1 parent dc6a1ff commit 67d3fb6
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion docs/training.md
Original file line number Diff line number Diff line change
Expand Up @@ -193,10 +193,12 @@ intended for training adapters. The `Trainer` class should still be used to full
class, simply initialize it the same way you would initialize the `Trainer` class, e.g.:

```python
from transformers.training_args import TrainingArguments

model.add_adapter(task_name)
model.train_adapter(task_name)

trainings_args = TrainingsArguments(
training_args = TrainingArguments(
learning_rate=1e-4,
num_train_epochs=6,
)
Expand Down

0 comments on commit 67d3fb6

Please sign in to comment.