Add example inference config yaml
Browse files- inference_config.yaml +22 -0
    	
        inference_config.yaml
    ADDED
    
    | @@ -0,0 +1,22 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            experiment_dir: /output_directory
         | 
| 2 | 
            +
            n_forward_steps: 400 
         | 
| 3 | 
            +
            forward_steps_in_memory: 50
         | 
| 4 | 
            +
            checkpoint_path: /ace2_era5_ckpt.tar
         | 
| 5 | 
            +
            logging:
         | 
| 6 | 
            +
              log_to_screen: true
         | 
| 7 | 
            +
              log_to_wandb: false
         | 
| 8 | 
            +
              log_to_file: true
         | 
| 9 | 
            +
              project: ace
         | 
| 10 | 
            +
            initial_condition:
         | 
| 11 | 
            +
              path: /initial_conditions/ic_2020.nc
         | 
| 12 | 
            +
              start_indices:
         | 
| 13 | 
            +
                times:
         | 
| 14 | 
            +
                  - "2020-01-01T00:00:00"
         | 
| 15 | 
            +
            forcing_loader:
         | 
| 16 | 
            +
              dataset:
         | 
| 17 | 
            +
                data_path: /forcing_data
         | 
| 18 | 
            +
              num_data_workers: 4
         | 
| 19 | 
            +
            data_writer:
         | 
| 20 | 
            +
              save_prediction_files: true
         | 
| 21 | 
            +
              save_monthly_files: false
         | 
| 22 | 
            +
              names: ["TMP2m", "VGRD10m", "PRATEsfc"]
         | 
