Initial model upload plus config and scalers.

config.yaml

@@ -0,0 +1,31 @@
+model:
+  img_size: 4096
+  patch_size: 16
+  embed_dim: 1280
+  depth: 10
+  n_spectral_blocks: 2
+  num_heads: 16
+  mlp_ratio: 4.0
+  drop_rate: 0.0
+  window_size: 2
+  dp_rank: 4
+  learned_flow: false
+  use_latitude_in_learned_flow: false
+  rpe: false
+  ensemble: null
+  finetune: false
+data:
+  sdo_channels: [
+    "aia94",  "aia131", "aia171", "aia193",
+    "aia211", "aia304", "aia335", "aia1600",
+    "hmi_m",  "hmi_bx", "hmi_by", "hmi_bz",  "hmi_v",
+  ]
+  time_delta_input_minutes: [-60, 0]
+  time_delta_target_minutes: +60
+  n_input_timestamps: 2
+  pooling: 1
+  random_vert_flip: false
+  drop_hmi_probablity: 0.0
+  num_mask_aia_channels: 0
+  use_latitude_in_learned_flow: false
+

scalers.yaml

@@ -0,0 +1,169 @@
+aia94:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 5.104914665222168
+  mean: 0.007682247087359428
+  min: 0.0
+  ssq: 8766571.0
+  std: 0.014216765761375427
+  sum: 257902320.0
+  sl_scale_factor: 1.0e-02
+aia131:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 4.642958164215088
+  mean: 0.03911230340600014
+  min: 0.0
+  ssq: 153411952.0
+  std: 0.05513598397374153
+  sum: 1313047296.0
+  sl_scale_factor: 1.0e-02
+aia171:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 4.716078758239746
+  mean: 0.6208767294883728
+  min: 0.0
+  ssq: 21844160512.0
+  std: 0.5149681568145752
+  sum: 20843583488.0
+  sl_scale_factor: 1.0e-02
+aia193:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 5.091572284698486
+  mean: 0.6960407495498657
+  min: 0.0
+  ssq: 27141281792.0
+  std: 0.569206714630127
+  sum: 23366930432.0
+  sl_scale_factor: 1.0e-02
+aia211:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 5.104914665222168
+  mean: 0.3032558560371399
+  min: 0.0
+  ssq: 6964888576.0
+  std: 0.3398558497428894
+  sum: 10180665344.0
+  sl_scale_factor: 1.0e-02
+aia304:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 5.104914665222168
+  mean: 0.24362024664878845
+  min: 0.0
+  ssq: 3977180160.0
+  std: 0.24314379692077637
+  sum: 8178626048.0
+  sl_scale_factor: 1.0e-02
+aia335:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 5.104914665222168
+  mean: 0.030005566775798798
+  min: 0.0
+  ssq: 101809864.0
+  std: 0.04617703706026077
+  sum: 1007323136.0
+  sl_scale_factor: 1.0e-02
+aia1600:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 5.104914665222168
+  mean: 0.27640828490257263
+  min: 0.0
+  ssq: 5027391488.0
+  std: 0.2708348035812378
+  sum: 9279360000.0
+  sl_scale_factor: 1.0e-02
+hmi_bx:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 3.737321376800537
+  mean: 0.0010142404353246093
+  min: -3.7792491912841797
+  ssq: 3866785536.0
+  std: 0.3393826484680176
+  sum: 34049276.0
+  sl_scale_factor: 1.0e-02
+hmi_by:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 3.6676809787750244
+  mean: 0.0008488616440445185
+  min: -3.9032106399536133
+  ssq: 3804123904.0
+  std: 0.33662179112434387
+  sum: 28497312.0
+  sl_scale_factor: 1.0e-02
+hmi_bz:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 3.9140491485595703
+  mean: -6.594143087568227e-06
+  min: -3.9124834537506104
+  ssq: 323843296.0
+  std: 0.09821633249521255
+  sum: -221373.359375
+  sl_scale_factor: 1.0e-02
+hmi_m:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 3.6454217433929443
+  mean: -1.8074297258863226e-05
+  min: -3.5852091312408447
+  ssq: 327147904.0
+  std: 0.09871617704629898
+  sum: -606776.0
+  sl_scale_factor: 1.0e-02
+hmi_v:
+  base: surya.datasets.transformations
+  class: StandardScaler
+  count: 33571209216
+  epsilon: 1.0e-08
+  is_fitted: 'true'
+  max: 4.553315162658691
+  mean: 0.028541626408696175
+  min: -4.583911418914795
+  ssq: 130445197312.0
+  std: 1.9709930419921875
+  sum: 958176896.0
+  sl_scale_factor: 1.0e-02

surya_366m.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7cac7599a371a3a1c61ba133aa192db987fa9efb0ab88a0ccc3119220c498bf1
+size 1800384309