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

1 maxpool layer