def __init__(self):
super(SimpleConv, self).__init__()
ConvNet.add_conv(self, 1, 30, 5, 5)
ConvNet.add_batch_normalization(self, 30 * 24 * 24, "Relu")
ConvNet.add_pooling(self, 2, 2, stride=2)
ConvNet.add_affine(self, 30 * 12 * 12, 200)
ConvNet.add_batch_normalization(self, 200, "Relu")
ConvNet.add_affine(self, 200, 10)
ConvNet.add_softmax(self)
def __init__(self):
super(SimpleConv, self).__init__()
ConvNet.add_conv(self, 3, 64, 3, 3, pad=1)
ConvNet.add_batch_normalization(self, 64 * 96 * 96, "Relu")
ConvNet.add_pooling(self, 2, 2, stride=2)
ConvNet.add_conv(self, 64, 16, 3, 3, pad=1)
ConvNet.add_batch_normalization(self, 16 * 48 * 48, "Relu")
ConvNet.add_pooling(self, 2, 2, stride=2)
ConvNet.add_affine(self, 16 * 24 * 24, 200)
ConvNet.add_batch_normalization(self, 200, "Relu")
ConvNet.add_affine(self, 200, 2)
ConvNet.add_softmax(self)