config.json

{
    "model_type": "Pix2Pix",
    "generator": {
        "in_channels": 3,
        "out_channels": 3,
        "layers": [
            64,
            128,
            256,
            512
        ]
    },
    "input_size": [
        256,
        256
    ],
    "task": "reflection_removal"
}