hiro_downscaling_ace2s_global_output.yaml

experiment_dir: /output_directory
patch:
  divide_generation: true
  composite_prediction: true
  coarse_horizontal_overlap: 1
model:
  checkpoint_path: /HiRO.ckpt
  rename:
    eastward_wind_at_ten_meters: UGRD10m
    northward_wind_at_ten_meters: VGRD10m
data:
  coarse:
  - data_path: /output_directory
    engine: zarr
    file_pattern: output_6hourly_ace2s_predictions_ic0000.zarr
  batch_size: 4
  num_data_workers: 2
  strict_ensemble: False
outputs:
- name: "Global_HiRO_2023"
  save_vars: ["PRATEsfc"]
  n_ens: 1
  max_samples_per_gpu: 32
  # Note: for reference, 1 year of global 3km outputs is ~300 Gb per ensemble member
  time_range:
    start_time: "2023-01-01T00:00:00"
    stop_time: "2023-01-08T00:00:00"
  lat_extent:
    start: -66.0
    stop: 70.0
  lon_extent:
    start: 0
    stop: 360
logging:
  log_to_screen: true
  log_to_wandb: false
  log_to_file: true