Method __init__

models/networks.py:197–201

(self, pretrain=False)

class SimpleResNetEncoder4(nn.Module):
    def __init__(self, pretrain=False):
        super().__init__()
        self.resnet_fpn = resnet_fpn_backbone('resnet34', pretrained=pretrain,
                                              norm_layer=None, returned_layers=[1, 2],
                                              trainable_layers=5)

    def forward(self, x):
        x = self.resnet_fpn(x)