{
    "batch_size": 1,
    "data": {
        "datasets": [
            {
                "dataset": {
                    "domain": "a",
                    "name": "toyzero-precropped-v1",
                    "path": "slats/slats_tiles/"
                },
                "shape": [
                    1,
                    256,
                    256
                ],
                "transform_test": null,
                "transform_train": null
            },
            {
                "dataset": {
                    "domain": "b",
                    "name": "toyzero-precropped-v1",
                    "path": "slats/slats_tiles/"
                },
                "shape": [
                    1,
                    256,
                    256
                ],
                "transform_test": null,
                "transform_train": null
            }
        ],
        "merge_type": "unpaired",
        "workers": null
    },
    "generator": {
        "model": "vit-unet",
        "model_args": {
            "activ": "gelu",
            "activ_output": null,
            "embed_features": 384,
            "features": 384,
            "ffn_features": 1536,
            "n_blocks": 12,
            "n_heads": 6,
            "norm": "layer",
            "rezero": true,
            "unet_activ": "leakyrelu",
            "unet_downsample": "conv",
            "unet_features_list": [
                48,
                96,
                192,
                384
            ],
            "unet_norm": null,
            "unet_upsample": "upsample-conv"
        }
    }
}