Method __init__

models/networks.py:142–145

(self, pretrain=False, use_layer2=False)

class SimpleResNetEncoder(nn.Module):
    def __init__(self, pretrain=False, use_layer2=False):
        super().__init__()
        self.resnet = models.resnet18(pretrained=pretrain)
        self.use_layer2 = use_layer2

    def forward(self, x):
        x = self.resnet.conv1(x)