%matplotlib inline
from fastai.gen_doc.nbdoc import *
from fastai.vision import *
from fastai.callbacks import *

path = untar_data(URLs.MNIST_SAMPLE)
data = ImageDataBunch.from_folder(path)
def simple_learner(): return Learner(data, simple_cnn((3,16,16,2)), metrics=[accuracy])
learn = simple_learner()

show_doc(Learner.lr_find)

learn.lr_find(stop_div=False, num_it=200)

learn.recorder.plot()

simple_learner().fit(2, 1e-2)

learn = simple_learner()
simple_learner().fit(2, 1e-0)

learn = simple_learner()
simple_learner().fit(2, 1e-3)

learn.lr_find(stop_div=False, num_it=200)

learn.recorder.plot(suggestion=True)

min_grad_lr = learn.recorder.min_grad_lr
min_grad_lr

learn = simple_learner()
simple_learner().fit(2, min_grad_lr)

show_doc(LRFinder)

show_doc(LRFinder.on_train_begin)

show_doc(LRFinder.on_batch_end)

show_doc(LRFinder.on_epoch_end)

show_doc(LRFinder.on_train_end)