/CIFAR10_NN

Neural Network model on the CIFAR-10 image dataset trained with mini-batch SGD.

Primary LanguagePython

CIFAR10_NN

Neural Network model on the CIFAR-10 image dataset trained with mini-batch SGD. Test accuracy of 41.3% with a fully connected 2 layer network (1024, 128, 128, 10). No data preprocessing and images converted to grayscale.