diff --git a/examples/MNIST.jl b/examples/MNIST.jl index e768e166..786d9293 100644 --- a/examples/MNIST.jl +++ b/examples/MNIST.jl @@ -1,5 +1,5 @@ using Flux, MNIST -using Flux: accuracy, onehot +using Flux: accuracy, onehot, tobatch data = [(trainfeatures(i), onehot(trainlabel(i), 0:9)) for i = 1:60_000] train = data[1:50_000]