| { |
| "model_name": "AtmoRep", |
| "model_type": "atmorep", |
| "architectures": [ |
| "TinyAtmoRep" |
| ], |
| "framework": "PyTorch", |
| "domain": "atmosphere", |
| "task": "masked-atmospheric-representation-learning", |
| "implementation": { |
| "entry_point": "model/tiny_atmorep.py", |
| "scope": "single-field compact validation model; official vorticity resources are retained separately" |
| }, |
| "architecture": { |
| "family": "masked-token transformer", |
| "input_shape": [ |
| 4, |
| 1, |
| 8, |
| 8 |
| ], |
| "input_shape_order": [ |
| "time", |
| "variable", |
| "latitude", |
| "longitude" |
| ], |
| "patch_shape": [ |
| 1, |
| 4, |
| 4 |
| ], |
| "embedding_size": 32, |
| "encoder_layers": 2, |
| "attention_heads": 4, |
| "supported_fields": 1 |
| }, |
| "official_resource": { |
| "field": "vorticity", |
| "model_levels": [ |
| 96, |
| 105, |
| 114, |
| 123, |
| 137 |
| ], |
| "global_grid_shape": [ |
| 721, |
| 1440 |
| ], |
| "spatial_resolution_degrees": 0.25, |
| "encoder_layers": 10, |
| "encoder_attention_heads": 16, |
| "decoder_layers": 10, |
| "decoder_attention_heads": 16 |
| }, |
| "configuration_sources": [ |
| "model/tiny_atmorep.py", |
| "resources/id4nvwbetz/model_id4nvwbetz.json", |
| "README.md" |
| ] |
| } |
|
|
