Closed
Description
In the tutorial of neural networks, the image of the example convnet is not the same as the code below it.
Based on the image conv layers should have 5x5 square kernels:
self.conv1 = nn.Conv2d(1, 6, 5)
self.conv2 = nn.Conv2d(6, 16, 5)
self.fc1 = nn.Linear(16 * 5 * 5, 120)
self.fc2 = nn.Linear(120, 84)
self.fc3 = nn.Linear(84, 10)
Metadata
Metadata
Assignees
Labels
No labels