Samudra/configs/fomo_om4/eval_cross_resolution.yaml at ce6603116ce0fbe8bb2f0d54c9aaa3ecffa7933c · m2lines/Samudra · GitHub

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
# yaml-language-server: $schema=../schemas/EvalConfig.json
#
# Cross-resolution eval for the FOMO model (KR2, issue #615).
#
# The rollout starts at ¼° and consumes 1° boundary forcings at each step.
# Prognostics stay at ¼° throughout the rollout — only the boundaries come
# from a coarser source. Enabled by the dual-perceiver encoder (PR #702),
# which produces the same latent grid for both streams when patch_extent is
# in degrees.
#
# As in eval_multiscale.yaml, all three resolutions must appear in
# `sources` so FOMOConfig.build computes `max_patch_size` correctly for the
# Perceiver's Fourier frequency range. The first source is the one used for
# inference (¼° here), and `boundary_source` specifies where to load
# boundary forcings from (1° here).

debug: false
save_zarr: true
disk_mode: true
inference_time:
  start: "2014-10-05"
  end: "2015-10-05"
num_model_steps_forward: 25

experiment:
  name: fomo_om4_eval_cross_resolution
  rand_seed: 15
  base_output_dir: .LOCAL
  wandb:
    mode: disabled
    project: default
  prognostic_vars_key: thermo_dynamic_all
  boundary_vars_key: tau_hfds_hfds_anom

data:
  static_data_vars: []
  concurrent_compute: true
  sources:
    # First source is used for inference prognostics.
    - data_location: om4_quarterdeg_v2/OM4.zarr
      data_means_location: om4_quarterdeg_v2/OM4_means.zarr
      data_stds_location: om4_quarterdeg_v2/OM4_stds.zarr
    - data_location: om4_halfdeg_v4/OM4.zarr
      data_means_location: om4_halfdeg_v4/OM4_means.zarr
      data_stds_location: om4_halfdeg_v4/OM4_stds.zarr
    - data_location: om4_onedeg_v3/OM4.zarr
      data_means_location: om4_onedeg_v3/OM4_means.zarr
      data_stds_location: om4_onedeg_v3/OM4_stds.zarr
  # Boundary forcings come from the 1° source, while prognostics come from
  # `sources[0]` (¼°). This is the KR2 asymmetric-inference setup.
  boundary_source:
    data_location: om4_onedeg_v3/OM4.zarr
    data_means_location: om4_onedeg_v3/OM4_means.zarr
    data_stds_location: om4_onedeg_v3/OM4_stds.zarr

model: !include model.yaml