def METHOD_NAME(model_dir, pretrained=False):
    model = SketchSimplification(
        mean=0.9664114577640158, std=0.0858381272736797)
    if pretrained:
        model.load_state_dict(
            torch.load(
                os.path.join(model_dir, 'sketch_simplification_gan.pth'),
                map_location='cpu'))
    return model