from fastai.gen_doc.nbdoc import *
from fastai.callbacks.fp16 import *
from fastai import *
from fastai.vision import *

show_doc(Learner.to_fp16)

path = untar_data(URLs.MNIST_SAMPLE)
data = ImageDataBunch.from_folder(path)
model = simple_cnn((3,16,16,2))
learn = Learner(data, model, metrics=[accuracy]).to_fp16()
learn.fit_one_cycle(1)

show_doc(MixedPrecision)

show_doc(MixedPrecision.on_backward_begin)

show_doc(MixedPrecision.on_backward_end)

show_doc(MixedPrecision.on_loss_begin)

show_doc(MixedPrecision.on_step_end)

show_doc(MixedPrecision.on_train_begin)

show_doc(MixedPrecision.on_train_end)