library(fastai)
library(magrittr)
data = Data_Loaders(train_loader, test_loader)
nn = nn()
learn = Learner(data, Net(), loss_func = nn$functional$nll_loss,
opt_func = Adam(), metrics = accuracy, cbs = CudaCallback())
learn %>% fit_one_cycle(1, 1e-2)