def __init__(self, n_out_channels, kernel_size, stride, pad, group):
self.n_out_channels = n_out_channels
self.kernel_size = kernel_size
self.stride = stride
self.pad = pad
self.group = group
# Parameters
self.params = OrderedDict()
self.params["w"] = Variable()
self.params["b"] = Variable()
super(ConvLayer, self).__init__()
def __init__(self, n_out, init_type):
self.n_out = n_out
self.init_type = init_type
self.params = OrderedDict()
self.params["w"] = Variable()
self.params["b"] = Variable()