def __init__(self, in_channels=1, n_hidden=100, n_outputs=10): FunctionSet.__init__( self, conv1=F.Convolution2D(in_channels, 32, 5), conv2=F.Convolution2D(32, 32, 5), l3=F.Linear(288, n_hidden), l4=F.Linear(n_hidden, n_outputs) )