al-ilm-nn-julia-flux-6.jl

using Flux: ADAM,
            argmax, 
            Chain,
            Dense, 
            params,
            relu, 
            softmax

m = Chain(
    Dense(32^2 * 3, 32 * 10, relu),
    Dense(32 * 10, 100),
    softmax
);

loss(x, y) = crossentropy(m(x), y);
accuracy(x, y) = mean(argmax(m(x)) .== argmax(y));