Hugging Face's logo

Francismj
/
sonny

Other
PyTorch
Safetensors
weather
era5
sonny
stepnets
Model card Files
xet
 Community
sonny
File size: 2,583 Bytes
1fdde4f
6aabbcd
1fdde4f
6aabbcd
 
c32be8e
1fdde4f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "model_name": "Sonny",
  "sonny_version": "1.0",
  "architecture_name": "Sonny",
  "architecture_family": "StepNets",
  "class_qualname": "networks.sonny.Sonny",
  "source_checkpoint": "stormbreaker5_checkpoint_ema_epoch_050.pth",
  "source_epoch": 50,
  "init_args": {
    "variables": [
      "2m_temperature",
      "10m_u_component_of_wind",
      "10m_v_component_of_wind",
      "mean_sea_level_pressure",
      "geopotential_50",
      "geopotential_100",
      "geopotential_150",
      "geopotential_200",
      "geopotential_250",
      "geopotential_300",
      "geopotential_400",
      "geopotential_500",
      "geopotential_600",
      "geopotential_700",
      "geopotential_850",
      "geopotential_925",
      "geopotential_1000",
      "u_component_of_wind_50",
      "u_component_of_wind_100",
      "u_component_of_wind_150",
      "u_component_of_wind_200",
      "u_component_of_wind_250",
      "u_component_of_wind_300",
      "u_component_of_wind_400",
      "u_component_of_wind_500",
      "u_component_of_wind_600",
      "u_component_of_wind_700",
      "u_component_of_wind_850",
      "u_component_of_wind_925",
      "u_component_of_wind_1000",
      "v_component_of_wind_50",
      "v_component_of_wind_100",
      "v_component_of_wind_150",
      "v_component_of_wind_200",
      "v_component_of_wind_250",
      "v_component_of_wind_300",
      "v_component_of_wind_400",
      "v_component_of_wind_500",
      "v_component_of_wind_600",
      "v_component_of_wind_700",
      "v_component_of_wind_850",
      "v_component_of_wind_925",
      "v_component_of_wind_1000",
      "temperature_50",
      "temperature_100",
      "temperature_150",
      "temperature_200",
      "temperature_250",
      "temperature_300",
      "temperature_400",
      "temperature_500",
      "temperature_600",
      "temperature_700",
      "temperature_850",
      "temperature_925",
      "temperature_1000",
      "specific_humidity_50",
      "specific_humidity_100",
      "specific_humidity_150",
      "specific_humidity_200",
      "specific_humidity_250",
      "specific_humidity_300",
      "specific_humidity_400",
      "specific_humidity_500",
      "specific_humidity_600",
      "specific_humidity_700",
      "specific_humidity_850",
      "specific_humidity_925",
      "specific_humidity_1000"
    ],
    "in_img_size": [
      120,
      240
    ],
    "patch_size": 2,
    "use_cnn_head": true,
    "cnn_head_channels": 256,
    "hidden_size": 384,
    "depth": 12,
    "num_heads": 6,
    "mlp_ratio": 4.0,
    "step_ratio": 0.5
  }
}