📝
Posts tagged with :memo:
NetC
(15 points) This third neural network will be a convolutional neural network. It should use the following operations in this order:
Use a convolution layer with kernel-width 5 and depth 25 (note: zero padding and stride = 1 are by default in `nn.Conv2d`)
Use the ReLU activation function
Use a max-pool operation with kernel-width 2 and stride 2
Reshape the image to a vector for the next step's input using .view().
* Use a fully-connected linear layer to the 10 outputs.
This architecture can be seen visually in the following diagram (the left-most object is the input image).
Notice that these diagrams use the following notation to describe the dimensions of the results at each step.
channels @ height x width
