from fastai.basic_train import *
from fastai.gen_doc.nbdoc import *
from fastai.callbacks import *

from fastai.vision import *

path = untar_data(URLs.MNIST_SAMPLE)
data = ImageDataBunch.from_folder(path)

model = simple_cnn((3,16,16,2))

learn = Learner(data, model)

learn.fit(1)

learn.metrics=[accuracy]
learn.fit(1)

cb = OneCycleScheduler(learn, lr_max=0.01)
learn.fit(1, callbacks=cb)

learn.recorder.plot_lr(show_moms=True)

learn.fit_one_cycle(1)

learn = cnn_learner(data, models.resnet18, metrics=accuracy)
learn.fit_one_cycle(1)