def forward()

in visu/activ-retrieval.py [0:0]

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

def forward(model, my_layer, x):
    if model.sobel is not None:
        x = model.sobel(x)
    layer = 1
    res = {}
    for m in model.features.modules():
        if not isinstance(m, nn.Sequential):
            x = m(x)
            if isinstance(m, nn.ReLU):
                if layer == my_layer:
                    for channel in range(int(x.size()[1])):
                        key = 'layer' + str(layer) + '-channel' + str(channel)
                        res[key] = torch.squeeze(x.mean(3).mean(2))[:, channel]
                    return res
                layer = layer + 1
    return res