def forward()

in code/colored_mnist/main.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

    def forward(self, input):
      if flags.grayscale_model:
        out = input.view(input.shape[0], 2, 14 * 14).sum(dim=1)
      else:
        out = input.view(input.shape[0], 2 * 14 * 14)
      out = self._main(out)
      return out