From 2a703773aa86ee1b2d225b93a59b884a900f6330 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Wed, 20 Jan 2021 12:17:40 -0500 Subject: [PATCH] Fix style --- src/transformers/training_args.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py index c9d436ebc0..d725ef1cf9 100644 --- a/src/transformers/training_args.py +++ b/src/transformers/training_args.py @@ -106,7 +106,7 @@ class TrainingArguments: learning_rate (:obj:`float`, `optional`, defaults to 5e-5): The initial learning rate for :class:`~transformers.AdamW` optimizer. weight_decay (:obj:`float`, `optional`, defaults to 0): - The weight decay to apply (if not zero) to all layers except all bias and LayerNorm weights in + The weight decay to apply (if not zero) to all layers except all bias and LayerNorm weights in :class:`~transformers.AdamW` optimizer. adam_beta1 (:obj:`float`, `optional`, defaults to 0.9): The beta1 hyperparameter for the :class:`~transformers.AdamW` optimizer.