DL with fastai vision_learner
from fastai.vision.all import *
path = untar_data(URLs.MNIST_SAMPLE)
Path.BASE_PATH = path
path.ls()
dls = ImageDataLoaders.from_folder(path)
learn = vision_learner(dls, resnet18, pretrained=False, loss_func=nn.CrossEntropyLoss(), metrics=accuracy)
learn.fit_one_cycle(1, 0.1)