Hi, till now I was using small model for MNIST like this -
Input tensor was (64x1x28x28) and I used this for flatten images to vector. So it was (64x784)
images = images.view(images.shape, -1)
input_size = 784 hidden_sizes = [128, 64] output_size = 10 model = nn.Sequential(nn.Linear(input_size, hidden_sizes), nn.ReLU(), nn.Linear(hidden_sizes, hidden_sizes), nn.ReLU(), nn.Linear(hidden_sizes, output_size), nn.LogSoftmax(dim=1)) print(model)
I changed model to this bigger model That starts with these layers.
How should I change input?
I have (64x1x28x28), how should i transofrm it for this model ?
self.conv1 = nn.Conv2d(1, 32, kernel_size=3, stride=2, padding=1) self.activation1 = nn.ReLU() self.conv2 = nn.Conv2d(32, 64, kernel_size=3, stride=2, padding=1) self.activation2 = nn.ReLU() ETC....... AT THE END I HAVE self.linear1 = nn.Linear(128 * 5 * 5, 10)