# Paper-faithful configuration for arXiv:2302.01178, Table 12 and Appendix C.
experiment:
  name: cno_navier_stokes_2d
  paper: https://arxiv.org/pdf/2302.01178
  seed: 0                    # Paper-unspecified, explicit reproducibility choice.
  deterministic: true

paths:
  data_dir: /public/share/sugonhpcapp01/onestore/onedatasets/RPB_CNO/data
  train_file: NavierStokes_64x64_IN.h5
  id_test_file: NavierStokes_64x64_IN.h5
  ood_test_file: NavierStokes_128x128_OUT.h5
  checkpoint: weight/best_model.pth
  results_dir: results

data:
  input_key: input
  output_key: output
  train:
    start: 0
    stop: 750
  validation:
    start: 768
    stop: 896
  test_id:
    start: 896
    stop: 1024
  test_ood:
    start: 0
    stop: 128

# The paper requires [0,1] training normalization and reuse at test time but
# does not publish its constants. These extrema were measured once from the
# supplied 64x64 ID benchmark and are fixed for every split, including OOD.
normalization:
  source: supplied_64x64_id_benchmark
  input_min: -1.4294605255126953
  input_max: 1.4294605255126953
  output_min: -2.0383081436157227
  output_max: 2.0602376461029053
  epsilon: 1.0e-12

model:
  in_channels: 1
  out_channels: 1
  base_width: 32             # d_e in Table 12.
  levels: 3                  # M in Table 12.
  bottleneck_residual_blocks: 8
  intermediate_residual_blocks: 1
  kernel_size: 3
  latent_channels: 64        # Paper-unspecified lift/project internal width.
  activation_upsampling_factor: 2  # N_sigma.
  filter_taps: 12            # N_tap.
  filter_half_width: 0.8     # c_h.
  cutoff_denominator: 2.0001
  leaky_relu_slope: 0.2      # Paper-unspecified, official supplemental fact.

training:
  epochs: 1000
  batch_size: 32
  num_workers: 4
  optimizer: Adam
  learning_rate: 0.001
  weight_decay: 1.0e-10
  scheduler: StepLR
  scheduler_step_size: 1
  scheduler_gamma: 0.98
  early_stopping_patience: 50
  log_interval: 5
  device: auto

inference:
  batch_size: 16
  num_workers: 2
  device: auto
  metric_epsilon: 1.0e-12

paper_reference:
  metric: relative_median_l1_percent
  id: 2.76
  ood: 7.04