search

abhishekkrthakur / bert-sentiment

MIT License
269 stars 101 forks source link

In file train.py line 13 (get_linear_schedule_with_warmup import) #2

Closed VolodymyrGavrysh closed 4 years ago

VolodymyrGavrysh commented 4 years ago

In file train.py line 13 from transformers import get_linear_schedule_with_warmup

it is better to change to

from transformers import WarmupLinearSchedule as get_linear_schedule_with_warmup