From 24c5a6e351fe2f04ac79a0afd1a7a7811da6bd3c Mon Sep 17 00:00:00 2001 From: Ninnart Fuengfusin Date: Wed, 5 Aug 2020 20:34:57 +0900 Subject: [PATCH] Update optimization.py (#6261) --- src/transformers/optimization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/optimization.py b/src/transformers/optimization.py index f456f8ac2a..9419dc76c8 100644 --- a/src/transformers/optimization.py +++ b/src/transformers/optimization.py @@ -78,7 +78,7 @@ def get_linear_schedule_with_warmup(optimizer, num_warmup_steps, num_training_st num_warmup_steps (:obj:`int`): The number of steps for the warmup phase. num_training_steps (:obj:`int`): - The totale number of training steps. + The total number of training steps. last_epoch (:obj:`int`, `optional`, defaults to -1): The index of the last epoch when resuming training.