- self.conv1 = nn.Conv2d(1, 10, kernel_size=9)
- self.conv2 = nn.Conv2d(10, 20, kernel_size=6)
- self.fc1 = nn.Linear(500, 100)
- self.fc2 = nn.Linear(100, 2)
+ self.conv1 = nn.Conv2d(1, 6, kernel_size=21)
+ self.conv2 = nn.Conv2d(6, 16, kernel_size=19)
+ self.conv3 = nn.Conv2d(16, 120, kernel_size=18)
+ self.fc1 = nn.Linear(120, 84)
+ self.fc2 = nn.Linear(84, 2)