alecokas

# Julia version 1.4.1 and Flux version 0.10.4
using Flux
using Flux: logitcrossentropy, onehotbatch, onecold, @epochs
using Flux.Data: DataLoader
using MLDatasets: MNIST
using Random
using Zygote


function get_dataloaders(batch_size::Int, shuffle::Bool)