andreperkins commited on
Commit
fb3ec0b
·
verified ·
1 Parent(s): 43ed63d

Add sample inference files and config (#1)

Browse files

- Add sample inference files and config (a64c52417161c546e10cc6a35a4f3f3294a41c28)
- Update model checkpoint name (78b860352a1c7c344f6456420d327079b49e1fcf)

Files changed (2) hide show
  1. climSST.tar.gz +3 -0
  2. 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
+