Trains the model for few epochs, iteratively increasing learning rate. Tries to provide insight with respect to the choice of a learning rate by iteratively increasing learning rates from lr_min and lr_max, recording performance gains (c.f. Leslie N. Smith: Cyclical Learning Rates for Training Neural Networks 2015).
'find_lr' is also available via Learner$lr_find(...)
.
find_lr( learner, task, epochs = 5, lr_min = 10^-4, lr_max = 0.8, batch_size = 128L )
learner | |
---|---|
task | |
epochs |
|
lr_min |
|
lr_max |
|
batch_size |
|