Basic implimentation of a CNN using pytorch.
1 converlutional layer:
-> input depth 3
-> output depth 16
-> kernel size 3
-> padding 1

1 maxpool layer

This is a basic implementation of a neural network using pytorch. The network accepts an input of 784 and generate a final output of 10. eg(classifying 28x28 -> 784 images of 10 digits)