test model export

Test the Learner.export feature

from tempfile import TemporaryDirectory
from fastai.vision.all import *
from fastcore.test import *

def label_func(f): return f[0].isupper()
path = untar_data(URLs.PETS)
files = get_image_files(path/"images")
dls = ImageDataLoaders.from_name_func(path, files, label_func, item_tfms=Resize(32))

with TemporaryDirectory() as td:
    learn = vision_learner(dls, resnet18, metrics=error_rate, path=td)
    learn.fine_tune(1,base_lr=0.00001)
    learn.export("model.pkl")
    
    learn2 = load_learner(Path(td) / "model.pkl", cpu=False)

o1 = learn.predict(files[0])
o2 = learn2.predict(files[0])

test_eq(o1[:2],o2[:2])
test_close(o1[-1], o2[-1])

Other Formats