Skip to main content

Image Classification using CNN

Various configurations of CNN are used to classify images from CIFAR-10 dataset Model-1: SGD optimizer, learning rate = 0.001, momentum = 0.9, cross-entropy loss Model-2: Adam optimizer, learning rate = 0.01, cross-entropy loss Model-3: SGD optimizer, learning rate = 0.001, momentum = 0.9, squared-error loss Model-4: Adam optimizer, learning rate = 0.01, squared-error loss Model-5: Modified ResNet, SGD optimizer, learning rate = 0.01 , momentum = 0.9, cross-entropy loss.
Created on June 1|Last edited on June 1
CNN architecture used for Model-1 to Model-4: