config.json

{
  "architectures": {
    "base": "mae_vit_base_patch16",
    "large": "mae_vit_large_patch16"
  },
  "input_channels": 2,
  "input_size": [
    256,
    256
  ],
  "mask_ratio": 0.75,
  "norm_pix_loss": true,
  "normalization": {
    "mean": [
      -5.5445e-10,
      -5.5445e-10
    ],
    "type": "channel-wise standardization",
    "variance": [
      5.7916e-10,
      5.7916e-10
    ]
  },
  "patch_size": 16,
  "source_domain": "Sionna/3GPP simulated channel corpus"
}