def _forward_impl()

in pytouch/models/slip_resnet.py [0:0]

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

    def _forward_impl(self, x):
        # See note [TorchScript super()]

        B, C, T, W, H = x.size()[0], x.size()[1], x.size()[2], x.size()[3], x.size()[4]
        x = x.view([(B * T), C, W, H])

        x = self.conv1(x)
        x = self.bn1(x)
        x = self.relu(x)
        x = self.maxpool(x)

        x = self.layer1(x)
        x = self.layer2(x)
        x = self.layer3(x)
        x = self.layer4(x)

        x = self.avgpool(x)
        x = torch.flatten(x, 1)

        _, C = x.size()
        x = x.view([B, T, C])
        x = torch.mean(x, 1)

        x = self.fc(x)

        return x