Hi, I’m just getting started with the pytorch.Preformatted text In this example, I don’t understand why the input of the full connected layer is the output latitude of the previous convolution multiplied by the size of the convolution kernel? This is my code class net(nn.Module): def __init_…

The 5*5 are not derived from the kernel size, but the spatial size of the input to the linear layer, which is apparently also 5*5. You can add a print statement right before feeding the activation to self.fc1 to see the activation shape, which should be [batch_size, 16, 5, 5] before flattening.

PyTorch Forums

About nn.Linear(1655,120)

cclover November 26, 2019, 9:20am 4

Input size of fc layer in tutorial? vision

The input of a Pytorch Neural Network is of type [BATCH_SIZE] * [CHANNEL_NUMBER] * [HEIGHT] * [WIDTH]. Example : So lets assume you image is of dimension 1×3×32×32 meaning that you have 1 image with 3 channels (RGB) with height 32 and width 32. So using the formular of convolution which is ((W - F + 2P)/ S )+1 and ((H - F + 2P)/ S )+1 . The first one is for the Width and the second one is for the height NOTE : Delete the the 5th line of the code because you already have pooling in the 12…

This might help

show post in topic

Home
Categories
FAQ/Guidelines
Terms of Service
Privacy Policy

Powered by Discourse, best viewed with JavaScript enabled

About nn.Linear(16*5*5,120)

About nn.Linear(1655,120)