felix-laumann:

I’m using the same layer dimensions as in AlexNet and the input images have the dimension ([4, 3, 227, 227]) (batch size = 4), but make modifications in the convolutional layers. Precisely, I’m defining my own convolutional layer exactly as a nn.Conv2d , but the output of the first convolutional layer of “my” AlexNet has not the shape ([96, 3, 11, 11]) as it is supposed to have, but ([4, 96, 217, 217]) .


I am facing the same issue when using the F.conv2d layer from the functional module. Did you resolve this issue?