inference-config.yaml

experiment_dir: /results
n_coupled_steps: 2920 # 40 years
coupled_steps_in_memory: 5
checkpoint_path: /SamudrACE-E3SMv3.tar
data_writer:
  ocean:
    save_prediction_files: false
    save_monthly_files: false
  atmosphere:
    save_prediction_files: false
    save_monthly_files: false
logging:
  log_to_screen: true
  log_to_wandb: false # Set to true to log to wandb
  log_to_file: true
  project: WANDB-PROJECT
  entity: WANDB-ENTITY
forcing_loader:
  atmosphere:
    dataset:
      data_path: /forcing_data
      file_pattern: atmosphere-forcing-1yr.nc
      n_repeats: 43
  ocean:
    dataset:
      data_path: /forcing_data
      file_pattern: ocean-forcing-1yr.nc
      n_repeats: 43
  num_data_workers: 1
initial_condition:
  ocean:
    path: /initial_conditions/SamudrACE-E3SMv3-ICx3-train_ocean_ic.nc
  atmosphere:
    path: /initial_conditions/SamudrACE-E3SMv3-ICx3-train_atmosphere_ic.nc
  start_indices:
    first: 0
    n_initial_conditions: 3