kashif HF Staff commited on
Commit
d5a5dc5
·
verified ·
1 Parent(s): dbdc061

Move mask

Browse files

Keep the mesh mask with preprocessing. Tensor values are unchanged; checked checkpoint roundtrips and local forecasts.

attention_mask.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d240bca4346a43d753c41b4c053483783e709fa5600e9b21faef552a21736479
3
+ size 22284
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:21480fcd5d25ca9272625b6441a0fbdd907db32979469926ac18f75448711784
3
- size 139728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85e94c7cfed21659a5f8e9e330c474e4e53a9601f41617cffb66882cc200afb2
3
+ size 117444
preprocessor_config.json CHANGED
@@ -7,119 +7,7 @@
7
  "vertical_velocity",
8
  "specific_humidity"
9
  ],
10
- "diffs_stddev_by_level": {
11
- "10m_u_component_of_wind": [
12
- 1.0
13
- ],
14
- "10m_v_component_of_wind": [
15
- 1.0
16
- ],
17
- "2m_temperature": [
18
- 1.0
19
- ],
20
- "cyclone_all_wind_disc": [
21
- 1.0
22
- ],
23
- "cyclone_exists_gaussian_unit_mode": [
24
- 1.0
25
- ],
26
- "cyclone_usa_pres_disc": [
27
- 1.0
28
- ],
29
- "cyclone_usa_r34_ne_radius_disc": [
30
- 1.0
31
- ],
32
- "cyclone_usa_r34_nw_radius_disc": [
33
- 1.0
34
- ],
35
- "cyclone_usa_r34_se_radius_disc": [
36
- 1.0
37
- ],
38
- "cyclone_usa_r34_sw_radius_disc": [
39
- 1.0
40
- ],
41
- "cyclone_usa_r50_ne_radius_disc": [
42
- 1.0
43
- ],
44
- "cyclone_usa_r50_nw_radius_disc": [
45
- 1.0
46
- ],
47
- "cyclone_usa_r50_se_radius_disc": [
48
- 1.0
49
- ],
50
- "cyclone_usa_r50_sw_radius_disc": [
51
- 1.0
52
- ],
53
- "cyclone_usa_r64_ne_radius_disc": [
54
- 1.0
55
- ],
56
- "cyclone_usa_r64_nw_radius_disc": [
57
- 1.0
58
- ],
59
- "cyclone_usa_r64_se_radius_disc": [
60
- 1.0
61
- ],
62
- "cyclone_usa_r64_sw_radius_disc": [
63
- 1.0
64
- ],
65
- "cyclone_usa_rmw_disc": [
66
- 1.0
67
- ],
68
- "cyclone_usa_wind_disc": [
69
- 1.0
70
- ],
71
- "day_progress_cos": [
72
- 1.0
73
- ],
74
- "day_progress_sin": [
75
- 1.0
76
- ],
77
- "geopotential": [
78
- 1.0,
79
- 1.0
80
- ],
81
- "geopotential_at_surface": [
82
- 1.0
83
- ],
84
- "land_sea_mask": [
85
- 1.0
86
- ],
87
- "mean_sea_level_pressure": [
88
- 1.0
89
- ],
90
- "sea_surface_temperature": [
91
- 1.0
92
- ],
93
- "specific_humidity": [
94
- 1.0,
95
- 1.0
96
- ],
97
- "temperature": [
98
- 1.0,
99
- 1.0
100
- ],
101
- "total_precipitation_6hr": [
102
- 1.0
103
- ],
104
- "u_component_of_wind": [
105
- 1.0,
106
- 1.0
107
- ],
108
- "v_component_of_wind": [
109
- 1.0,
110
- 1.0
111
- ],
112
- "vertical_velocity": [
113
- 1.0,
114
- 1.0
115
- ],
116
- "year_progress_cos": [
117
- 1.0
118
- ],
119
- "year_progress_sin": [
120
- 1.0
121
- ]
122
- },
123
  "feature_extractor_type": "WeatherNext2FeatureExtractor",
124
  "forcing_variables": [
125
  "year_progress_sin",
@@ -152,119 +40,7 @@
152
  "day_progress_sin",
153
  "day_progress_cos"
154
  ],
155
- "mean_by_level": {
156
- "10m_u_component_of_wind": [
157
- 0.0
158
- ],
159
- "10m_v_component_of_wind": [
160
- 0.0
161
- ],
162
- "2m_temperature": [
163
- 0.0
164
- ],
165
- "cyclone_all_wind_disc": [
166
- 0.0
167
- ],
168
- "cyclone_exists_gaussian_unit_mode": [
169
- 0.0
170
- ],
171
- "cyclone_usa_pres_disc": [
172
- 0.0
173
- ],
174
- "cyclone_usa_r34_ne_radius_disc": [
175
- 0.0
176
- ],
177
- "cyclone_usa_r34_nw_radius_disc": [
178
- 0.0
179
- ],
180
- "cyclone_usa_r34_se_radius_disc": [
181
- 0.0
182
- ],
183
- "cyclone_usa_r34_sw_radius_disc": [
184
- 0.0
185
- ],
186
- "cyclone_usa_r50_ne_radius_disc": [
187
- 0.0
188
- ],
189
- "cyclone_usa_r50_nw_radius_disc": [
190
- 0.0
191
- ],
192
- "cyclone_usa_r50_se_radius_disc": [
193
- 0.0
194
- ],
195
- "cyclone_usa_r50_sw_radius_disc": [
196
- 0.0
197
- ],
198
- "cyclone_usa_r64_ne_radius_disc": [
199
- 0.0
200
- ],
201
- "cyclone_usa_r64_nw_radius_disc": [
202
- 0.0
203
- ],
204
- "cyclone_usa_r64_se_radius_disc": [
205
- 0.0
206
- ],
207
- "cyclone_usa_r64_sw_radius_disc": [
208
- 0.0
209
- ],
210
- "cyclone_usa_rmw_disc": [
211
- 0.0
212
- ],
213
- "cyclone_usa_wind_disc": [
214
- 0.0
215
- ],
216
- "day_progress_cos": [
217
- 0.0
218
- ],
219
- "day_progress_sin": [
220
- 0.0
221
- ],
222
- "geopotential": [
223
- 0.0,
224
- 0.0
225
- ],
226
- "geopotential_at_surface": [
227
- 0.0
228
- ],
229
- "land_sea_mask": [
230
- 0.0
231
- ],
232
- "mean_sea_level_pressure": [
233
- 0.0
234
- ],
235
- "sea_surface_temperature": [
236
- 0.0
237
- ],
238
- "specific_humidity": [
239
- 0.0,
240
- 0.0
241
- ],
242
- "temperature": [
243
- 0.0,
244
- 0.0
245
- ],
246
- "total_precipitation_6hr": [
247
- 0.0
248
- ],
249
- "u_component_of_wind": [
250
- 0.0,
251
- 0.0
252
- ],
253
- "v_component_of_wind": [
254
- 0.0,
255
- 0.0
256
- ],
257
- "vertical_velocity": [
258
- 0.0,
259
- 0.0
260
- ],
261
- "year_progress_cos": [
262
- 0.0
263
- ],
264
- "year_progress_sin": [
265
- 0.0
266
- ]
267
- },
268
  "nan_fill_values": {},
269
  "num_input_timesteps": 2,
270
  "pressure_levels": [
@@ -275,119 +51,7 @@
275
  "geopotential_at_surface",
276
  "land_sea_mask"
277
  ],
278
- "stddev_by_level": {
279
- "10m_u_component_of_wind": [
280
- 1.0
281
- ],
282
- "10m_v_component_of_wind": [
283
- 1.0
284
- ],
285
- "2m_temperature": [
286
- 1.0
287
- ],
288
- "cyclone_all_wind_disc": [
289
- 1.0
290
- ],
291
- "cyclone_exists_gaussian_unit_mode": [
292
- 1.0
293
- ],
294
- "cyclone_usa_pres_disc": [
295
- 1.0
296
- ],
297
- "cyclone_usa_r34_ne_radius_disc": [
298
- 1.0
299
- ],
300
- "cyclone_usa_r34_nw_radius_disc": [
301
- 1.0
302
- ],
303
- "cyclone_usa_r34_se_radius_disc": [
304
- 1.0
305
- ],
306
- "cyclone_usa_r34_sw_radius_disc": [
307
- 1.0
308
- ],
309
- "cyclone_usa_r50_ne_radius_disc": [
310
- 1.0
311
- ],
312
- "cyclone_usa_r50_nw_radius_disc": [
313
- 1.0
314
- ],
315
- "cyclone_usa_r50_se_radius_disc": [
316
- 1.0
317
- ],
318
- "cyclone_usa_r50_sw_radius_disc": [
319
- 1.0
320
- ],
321
- "cyclone_usa_r64_ne_radius_disc": [
322
- 1.0
323
- ],
324
- "cyclone_usa_r64_nw_radius_disc": [
325
- 1.0
326
- ],
327
- "cyclone_usa_r64_se_radius_disc": [
328
- 1.0
329
- ],
330
- "cyclone_usa_r64_sw_radius_disc": [
331
- 1.0
332
- ],
333
- "cyclone_usa_rmw_disc": [
334
- 1.0
335
- ],
336
- "cyclone_usa_wind_disc": [
337
- 1.0
338
- ],
339
- "day_progress_cos": [
340
- 1.0
341
- ],
342
- "day_progress_sin": [
343
- 1.0
344
- ],
345
- "geopotential": [
346
- 1.0,
347
- 1.0
348
- ],
349
- "geopotential_at_surface": [
350
- 1.0
351
- ],
352
- "land_sea_mask": [
353
- 1.0
354
- ],
355
- "mean_sea_level_pressure": [
356
- 1.0
357
- ],
358
- "sea_surface_temperature": [
359
- 1.0
360
- ],
361
- "specific_humidity": [
362
- 1.0,
363
- 1.0
364
- ],
365
- "temperature": [
366
- 1.0,
367
- 1.0
368
- ],
369
- "total_precipitation_6hr": [
370
- 1.0
371
- ],
372
- "u_component_of_wind": [
373
- 1.0,
374
- 1.0
375
- ],
376
- "v_component_of_wind": [
377
- 1.0,
378
- 1.0
379
- ],
380
- "vertical_velocity": [
381
- 1.0,
382
- 1.0
383
- ],
384
- "year_progress_cos": [
385
- 1.0
386
- ],
387
- "year_progress_sin": [
388
- 1.0
389
- ]
390
- },
391
  "target_variables": [
392
  "temperature",
393
  "geopotential",
 
7
  "vertical_velocity",
8
  "specific_humidity"
9
  ],
10
+ "diffs_stddev_by_level": {},
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  "feature_extractor_type": "WeatherNext2FeatureExtractor",
12
  "forcing_variables": [
13
  "year_progress_sin",
 
40
  "day_progress_sin",
41
  "day_progress_cos"
42
  ],
43
+ "mean_by_level": {},
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
44
  "nan_fill_values": {},
45
  "num_input_timesteps": 2,
46
  "pressure_levels": [
 
51
  "geopotential_at_surface",
52
  "land_sea_mask"
53
  ],
54
+ "stddev_by_level": {},
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
55
  "target_variables": [
56
  "temperature",
57
  "geopotential",