Solved by ptrblck in post #2 
              
              
                You could use the MNIST example as your template and just change a few lines of code. 
Since the MNIST data comes as grayscale images in the resolution 28x28, you would need to repeat the channel to simulate an RGB image as well as resizing it to 224x224 to fit your SqueezeNet. 
Here is a small exam…