utterworks / fast-bert

Super easy library for BERT based NLP models
Apache License 2.0
1.85k stars 342 forks source link

Reconfigured schedule_type to return correct number of arguments #228

Closed lingdoc closed 4 years ago

lingdoc commented 4 years ago

Number of arguments in Transformers schedulers (https://huggingface.co/transformers/_modules/transformers/optimization.html#get_constant_schedule) depends on schedule type. Addresses #211 .