from fastai.gen_doc.nbdoc import *
from fastai.callbacks.mixup import *
from fastai.vision import *
from fastai 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.fit(8)

model = simple_cnn((3,16,16,2))
learner = Learner(data, model, metrics=[accuracy]).mixup()
learner.fit(8)

show_doc(MixUpCallback, doc_string=False)

show_doc(MixUpCallback.on_batch_begin, doc_string=False)

show_doc(MixUpLoss, doc_string=False, title_level=3)

show_doc(MixUpLoss.forward)