search

utterworks / fast-bert

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

Error when changing schedule_type to warmup_constant #211

Open 0xhesch opened 4 years ago

0xhesch commented 4 years ago

I can only get the schedule types warmup_cosine and warmup_linear to work. All other schedule types eg. warmup_constant fail with

got an unexpected keyword argument 'num_training_steps'

Can anyone reproduce this?

Might be related: https://github.com/huggingface/transformers/issues/1837#issuecomment-554616094