{
  "architectures": {
    "base": "mae_vit_base_patch16",
    "large": "mae_vit_large_patch16"
  },
  "contents": "model parameters only; no optimizer or scaler state",
  "input_channels": 2,
  "input_size": [
    256,
    256
  ],
  "mask_ratio": 0.75,
  "name": "CSI-MAE",
  "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_checkpoints": {
    "base": {
      "architecture": "mae_vit_base_patch16",
      "epoch": 300,
      "file": "checkpoint-300.pth",
      "sha256": "74a0cad91298d689056ac36228c1dee4c380bc403eb3ad05ae726890b4589c31"
    },
    "large": {
      "architecture": "mae_vit_large_patch16",
      "epoch": 300,
      "file": "checkpoint-300.pth",
      "sha256": "b043e04b9963937b1a9c8ae01543bcb33ae2fb09de2a104a57d19ca73b4532f0"
    }
  },
  "source_domain": "Sionna/3GPP simulated channel corpus",
  "weight_files": {
    "csi-mae-base.pth": {
      "bytes": 446705111,
      "format": "PyTorch checkpoint",
      "recommended": false,
      "sha256": "c1ac1b2cddd4cd41b7edc0970a34e34668d48dec26c73be4feec68ccde846694",
      "variant": "base"
    },
    "csi-mae-base.safetensors": {
      "bytes": 446651400,
      "format": "safetensors",
      "recommended": true,
      "sha256": "db7ad83987ae7f8a5f0b42da9ba1749f792a27aa37356c2f5ba94aaa09f44a46",
      "variant": "base"
    },
    "csi-mae-large.pth": {
      "bytes": 1317084827,
      "format": "PyTorch checkpoint",
      "recommended": false,
      "sha256": "c65dba88c94c584d42f7e3cdb3975840d8af273992f1a4f8352f7a4931b59f38",
      "variant": "large"
    },
    "csi-mae-large.safetensors": {
      "bytes": 1317001152,
      "format": "safetensors",
      "recommended": true,
      "sha256": "d3ea94ead704488b410b1e86ceaa23f8c6446dfad93af24827715d453c915f25",
      "variant": "large"
    }
  }
}