riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-30 17:52:35 +06:00
Code Issues Actions 30 Packages Projects Releases Wiki Activity

Update optimization.py (#6261)

Browse Source
This commit is contained in:
Ninnart Fuengfusin 2020-08-05 20:34:57 +09:00 committed by GitHub
parent ed6b8f3128
commit 24c5a6e351
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/optimization.py
View File

@ -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.