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)
model = simple_cnn((3,16,16,2))
learn = Learner(data, model, metrics=[accuracy])

learn.lr_find()

learn.recorder.plot()

learn.fit(2, 5e-2)

model = simple_cnn((3,16,16,2))
learn = Learner(data, model, metrics=[accuracy])
learn.fit(2, 5e-3)

model = simple_cnn((3,16,16,2))
learn = Learner(data, model, metrics=[accuracy])
learn.fit_one_cycle(2, 5e-2)

learn.recorder.plot_lr(show_moms=True)

show_doc(OneCycleScheduler)

show_doc(OneCycleScheduler.steps, doc_string=False)

show_doc(OneCycleScheduler.on_train_begin, doc_string=False)

show_doc(OneCycleScheduler.on_batch_end, doc_string=False)

show_doc(OneCycleScheduler.on_epoch_end)