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"]
|