def __init__()

in train-cnn/Container/scripts/multi-train.py [0:0]

• 20 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self):
        super().__init__()
        self.conv_layer_1 = torch.nn.Sequential(
          torch.nn.Conv2d(1, 64, 3, padding=1),
          torch.nn.ReLU(),
          torch.nn.BatchNorm2d(64),
          torch.nn.MaxPool2d(2))
        self.conv_layer_2 = torch.nn.Sequential(
          torch.nn.Conv2d(64, 512, 3, padding=1),
          torch.nn.ReLU(),
          torch.nn.BatchNorm2d(512),
          torch.nn.MaxPool2d(2))
        self.conv_layer_3 = torch.nn.Sequential(
          torch.nn.Conv2d(512, 512, kernel_size=3, padding=1),
          torch.nn.ReLU(),
          torch.nn.BatchNorm2d(512),
          torch.nn.MaxPool2d(2))
        self.classifier = torch.nn.Sequential(
          torch.nn.Flatten(),
          torch.nn.Linear(in_features=2048, out_features=2))