File size: 1,599 Bytes
a4f6738 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | {
"model_name": "FengWu",
"model_type": "fengwu",
"architectures": [
"Fengwu"
],
"framework": "PyTorch",
"domain": "atmosphere",
"task": "global-medium-range-weather-forecasting",
"implementation": {
"entry_point": "model/fengwu.py",
"scope": "repository multi-modal encoder-fuser-decoder implementation"
},
"architecture": {
"family": "multi-branch hierarchical transformer",
"input_grid_shape": [
721,
1440
],
"patch_size": [
4,
4
],
"embedding_size": 192,
"stage_attention_heads": [
6,
12,
12,
6
],
"fuser_window_size": [
2,
6,
12
],
"variable_branches": [
"surface",
"geopotential",
"relative_humidity",
"temperature",
"u_component_of_wind",
"v_component_of_wind"
],
"input_channels": 189,
"output_channels": 189
},
"data": {
"dataset": "ERA5",
"spatial_resolution_degrees": 0.25,
"time_step_hours": 6,
"pressure_levels_hpa": [
1,
2,
3,
5,
7,
10,
20,
30,
50,
70,
100,
125,
150,
175,
200,
225,
250,
300,
350,
400,
450,
500,
550,
600,
650,
700,
750,
775,
800,
825,
850,
875,
900,
925,
950,
975,
1000
],
"pressure_level_variable_count": 5,
"surface_variable_count": 4
},
"configuration_sources": [
"conf/config.yaml",
"model/fengwu.py"
]
}
|