https://medium.com/ai-in-plain-english/image-classification-on-cifar-10-ii-shallow-neural-network-941b670fbbd0
