Pytorch, Torch, Torchvision, Deep Learning, Python.
CHANNELS = [1, 16, 32, 64] def __init__(self): super(CNN, self).__init__() CHANNELS = self.CHANNELS self.layer1 = nn.Sequential(nn.Conv2d( in_channels=CHANNELS[0], out_channels=CHANNELS[1], kernel_size=4, padding=1), nn.BatchNorm2d(CHANNELS[1]), nn.ReLU(), nn.MaxPool2d(kernel_size=2, stride=2))