andreperkins
commited on
Add sample inference files and config (#1)
Browse files- Add sample inference files and config (a64c52417161c546e10cc6a35a4f3f3294a41c28)
- Update model checkpoint name (78b860352a1c7c344f6456420d327079b49e1fcf)
- climSST.tar.gz +3 -0
- inference-config.yaml +27 -0
climSST.tar.gz
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0403900c09b5d8ee40f8d4355e49e1023b0bd42070e2338356b501e416af5011
|
3 |
+
size 599707052
|
inference-config.yaml
ADDED
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
experiment_dir: inference_output
|
2 |
+
n_forward_steps: 400
|
3 |
+
forward_steps_in_memory: 50
|
4 |
+
checkpoint_path: ace_climSST_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: climSST/ic_2021.zarr
|
12 |
+
start_indices:
|
13 |
+
n_initial_conditions: 2
|
14 |
+
first: 0
|
15 |
+
interval: 3
|
16 |
+
engine: zarr
|
17 |
+
forcing_loader:
|
18 |
+
dataset:
|
19 |
+
data_path: climSST
|
20 |
+
file_pattern: forcing_2021.zarr
|
21 |
+
engine: zarr
|
22 |
+
n_repeats: 2 # use this to extend the 1-year of forcing data to desired length
|
23 |
+
num_data_workers: 2
|
24 |
+
data_writer:
|
25 |
+
save_prediction_files: false
|
26 |
+
save_monthly_files: false
|
27 |
+
|