Method __init__

SRGAN/loss.py:16–24

(self)

class VGGLoss(nn.Module):
    def __init__(self):
        super().__init__()
        #features map = [1, 512, 14, 14]
        self.vgg = vgg19(pretrained=True).features[:36].eval().to(config.DEVICE)
        print(self.vgg)
        self.loss = nn.MSELoss()

        for param in self.vgg.parameters():
            param.requires_grad = False

    def forward(self, input, target):
        vgg_input_features = self.vgg(input)