Full Model Emulation
annakwa commited on
Commit
34a9f3f
·
1 Parent(s): 915993e

update readme to match configs

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -21,13 +21,13 @@ HiRO (High Resolution Output) is a diffusion model which generates downscaled fi
21
 
22
  1. Download this repository. Optionally, you can just download a subset of the `forcing_data` and `initial_conditions` for the period you are interested in.
23
 
24
- 2. Update paths in the `inference_config.yaml`. Specifically, update `experiment_dir`, `checkpoint_path`, `initial_condition.path` and `forcing_loader.dataset.path`.
25
 
26
- 3. Install code dependencies with `pip install fme`.
27
 
28
- 4. Run ACE inference with `python -m fme.ace.inference inference_config.yaml`.
29
 
30
- 5. Update paths in the `downscaling_config.yaml`. Specifically, update `experiment_dir`, `model.checkpoint_path`, and `data.coarse`. `data.coarse` data path(s) should point to the saved ACE inference output from step 4.
31
 
32
 
33
  ### Strengths and weaknesses
 
21
 
22
  1. Download this repository. Optionally, you can just download a subset of the `forcing_data` and `initial_conditions` for the period you are interested in.
23
 
24
+ 2. Install code dependencies with `pip install fme`.
25
 
26
+ 3. Update paths in the ACE inference config file `ace2s_inference_config_global.yaml`. Specifically, update `experiment_dir`, `checkpoint_path`, `initial_condition.path` and `forcing_loader.dataset.path`.
27
 
28
+ 4. Update paths in the HiRO downscaling inference config file `hiro_downscaling_ace2s_pnw_output.yaml`. Specifically, update `experiment_dir`, `model.checkpoint_path`, and `data.coarse`. The directory path in `data.coarse` should point to the same ACE inference `experiment_dir` from step 3. Optionally, if you wish to change the region and/or time selection of the area(s) to downscale you may edit those in the downscaling config (see [downscalimg inference docs](https://ai2-climate-emulator.readthedocs.io/en/latest/downscaling_inference.html) for more details). An example of global downscaling is also provided in `hiro_downscaling_ace2s_global_output.yaml`.
29
 
30
+ 4. Run the script `run-hiro-ace.sh`.
31
 
32
 
33
  ### Strengths and weaknesses