Method __init__

evaluation/lpips.py:24–30

(self)

class AlexNet(nn.Module):
    def __init__(self):
        super().__init__()
        self.layers = models.alexnet(weights=models.AlexNet_Weights.IMAGENET1K_V1).features
        self.channels = []
        for layer in self.layers:
            if isinstance(layer, nn.Conv2d):
                self.channels.append(layer.out_channels)

    def forward(self, x):
        fmaps = []