- nn.Conv2d(1, 16, kernel_size = 5),
- nn.MaxPool2d(3), nn.ReLU(),
- nn.Conv2d(16, 32, kernel_size = 5),
- nn.MaxPool2d(2), nn.ReLU(),
+ nn.Conv2d(1, 16, kernel_size=5),
+ nn.MaxPool2d(3),
+ nn.ReLU(),
+ nn.Conv2d(16, 32, kernel_size=5),
+ nn.MaxPool2d(2),
+ nn.ReLU(),