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