| { |
| "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" |
| } |
| } |
| } |
|
|
