Pytorch, Torch, Torchvision, Deep Learning, Python.
self.layer2 = nn.Sequential(nn.Conv2d( in_channels=CHANNELS[1], out_channels=CHANNELS[2], kernel_size=4, padding=1), nn.BatchNorm2d(CHANNELS[2]), nn.ReLU(), nn.MaxPool2d(kernel_size=2, stride=2)) self.layer3 = nn.Sequential(nn.Conv2d( in_channels=CHANNELS[2], out_channels=CHANNELS[3],