INNER CODE UNIT · Python

x

Mukosame/Anime2Sketch · model.py:142

        x = self.mlp(x) + x
        return x


def create_model(model):
    """Create a model for anime2sketch
    hardcoding the options for simplicity
    """

    norm_layer = functools.partial(nn.InstanceNorm2d, affine=False, track_running_stats=False)
    net = UnetGenerator(3, 1, 8, 64, norm_layer=norm_layer, use_dropout=False)

    if model == 'default':
        ckpt = torch.load('weights/netG.pth')
        for key in list(ckpt.keys()):
            if 'module.' in key:
                ckpt[key.replace('module.', '')] = ckpt[key]
                del ckpt[key]

View source record →