Hugging Face's logo

kashif
/
weathernext2

Other
Transformers
Safetensors
weathernext2
weather
forecasting
climate
graph-neural-network
Model card Files
xet
 Community
weathernext2
File size: 2,991 Bytes
0a48607
 
 
 
 
 
 
 
 
 
 
 
 
873b0c6
0a48607
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
873b0c6
0a48607
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
873b0c6
0a48607
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
{
  "aggregate_normalization": 4.0,
  "architectures": [
    "WeatherNext2ForWeatherForecasting"
  ],
  "atmospheric_variables": [
    "temperature",
    "geopotential",
    "u_component_of_wind",
    "v_component_of_wind",
    "vertical_velocity",
    "specific_humidity"
  ],
  "attention_bandwidth": 10273,
  "attention_dropout": 0.0,
  "attention_k_hop": 32,
  "ball_query_radius_fraction": 0.6,
  "dtype": "float32",
  "edge_hidden_size": 32,
  "forcing_variables": [
    "year_progress_sin",
    "year_progress_cos",
    "day_progress_sin",
    "day_progress_cos"
  ],
  "global_variables": [
    "year_progress_sin",
    "year_progress_cos"
  ],
  "grid_latitudes": 721,
  "grid_longitudes": 1440,
  "hidden_act": "gelu_pytorch_tanh",
  "hidden_size": 768,
  "initializer_range": 0.02,
  "input_variables": [
    "temperature",
    "geopotential",
    "u_component_of_wind",
    "v_component_of_wind",
    "vertical_velocity",
    "specific_humidity",
    "2m_temperature",
    "mean_sea_level_pressure",
    "10m_v_component_of_wind",
    "10m_u_component_of_wind",
    "sea_surface_temperature",
    "100m_u_component_of_wind",
    "100m_v_component_of_wind",
    "geopotential_at_surface",
    "land_sea_mask",
    "year_progress_sin",
    "year_progress_cos",
    "day_progress_sin",
    "day_progress_cos"
  ],
  "intermediate_size": 3072,
  "layer_norm_eps": 1e-05,
  "mesh_splits": 6,
  "mlp_act": "silu",
  "model_type": "weathernext2",
  "noise_channels": 32,
  "num_attention_heads": 6,
  "num_grid_to_mesh_edges": 1618824,
  "num_hidden_layers": 24,
  "num_input_timesteps": 2,
  "pressure_levels": [
    50,
    100,
    150,
    200,
    250,
    300,
    400,
    500,
    600,
    700,
    850,
    925,
    1000
  ],
  "sigmoid_shifted_outputs": {
    "cyclone_exists_gaussian_unit_mode": 2.0
  },
  "static_variables": [
    "geopotential_at_surface",
    "land_sea_mask"
  ],
  "target_variables": [
    "temperature",
    "geopotential",
    "u_component_of_wind",
    "v_component_of_wind",
    "vertical_velocity",
    "specific_humidity",
    "2m_temperature",
    "mean_sea_level_pressure",
    "10m_v_component_of_wind",
    "10m_u_component_of_wind",
    "sea_surface_temperature",
    "100m_u_component_of_wind",
    "100m_v_component_of_wind",
    "total_precipitation_6hr",
    "cyclone_exists_gaussian_unit_mode",
    "cyclone_all_wind_disc",
    "cyclone_usa_wind_disc",
    "cyclone_usa_r34_ne_radius_disc",
    "cyclone_usa_r34_se_radius_disc",
    "cyclone_usa_r34_sw_radius_disc",
    "cyclone_usa_r34_nw_radius_disc",
    "cyclone_usa_r50_ne_radius_disc",
    "cyclone_usa_r50_se_radius_disc",
    "cyclone_usa_r50_sw_radius_disc",
    "cyclone_usa_r50_nw_radius_disc",
    "cyclone_usa_r64_ne_radius_disc",
    "cyclone_usa_r64_se_radius_disc",
    "cyclone_usa_r64_sw_radius_disc",
    "cyclone_usa_r64_nw_radius_disc",
    "cyclone_usa_rmw_disc",
    "cyclone_usa_pres_disc"
  ],
  "time_step_hours": 6,
  "transformers_version": "5.16.0.dev0"
}