from fastai.vision.all import * path = untar_data(URLs.PETS)/'images' def is_cat(x): return x[0].isupper() dls = ImageDataLoaders.from_name_func('.', get_image_files(path), valid_pct=0.2, seed=42, label_func=is_cat, item_tfms=Resize(192)) learn = vision_learner(dls, resnet18, metrics=error_rate) learn.fine_tune(3) learn.export('model.pkl')