#all_slow

from fastai2.vision.all import *

from migrating_pytorch import *

data = DataLoaders(train_loader, test_loader)
learn = Learner(data, Net(), loss_func=F.nll_loss, opt_func=Adam, metrics=accuracy, cbs=CudaCallback)

learn.fit_one_cycle(epochs, lr)