| 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: | |
| # Save an ensemble of downscaled outputs over the 2023 period | |
| - name: "PNW_ensemble_2023" | |
| save_vars: ["PRATEsfc"] | |
| n_ens: 4 | |
| max_samples_per_gpu: 32 | |
| time_range: | |
| start_time: "2023-01-01T00:00:00" | |
| stop_time: "2023-12-31T18:00:00" | |
| lat_extent: | |
| start: 32.9 | |
| stop: 50.0 | |
| lon_extent: | |
| start: 233.0 | |
| stop: 250.0 | |
| # Example of a single event downscaling (exact time may not be interesting) | |
| - name: "WA_snapshot_20230206" | |
| save_vars: ["PRATEsfc"] | |
| n_ens: 16 | |
| max_samples_per_gpu: 8 | |
| event_time: "2023-02-06T06:00:00" | |
| lat_extent: | |
| start: 36.0 | |
| stop: 52.0 | |
| lon_extent: | |
| start: 228.0 | |
| stop: 244.0 | |
| logging: | |
| log_to_screen: true | |
| log_to_wandb: false | |
| log_to_file: true | |