Inference configs and initial conditions

Inference configuration files for Ai2 ace codebase, for 1% forcing, piControl forcing and present-day (OM4) forcing. The initial conditions for each forcing test are also included.

.gitattributes

@@ -35,3 +35,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 Arctic_FloeNet_2007.mp4 filter=lfs diff=lfs merge=lfs -text
 Arctic_FloeNet_2007.gif filter=lfs diff=lfs merge=lfs -text
+CM4_1pct_initial_condition.nc filter=lfs diff=lfs merge=lfs -text
+CM4_piControl_initial_condition.nc filter=lfs diff=lfs merge=lfs -text
+OM4_initial_condition.nc filter=lfs diff=lfs merge=lfs -text

CM4_1pct_initial_condition.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f4f2b2faafc09d9917225b6605ed11a4b8154168164423e6f63ec8826d54de62
+size 4178960

CM4_piControl_initial_condition.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbe238de1b3b13642a4ca3eb38a4ac914e14a18822a07db59748d21381ed0085
+size 4178965

OM4_initial_condition.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71e7b33d99094dda5bef564385cdf6abb5f240c16ed8c72637e6a688f37389ed
+size 4176264

inference_config_1pct.yaml

@@ -0,0 +1,19 @@
+experiment_dir: ./inference/FloeNet_1pct_forcing
+n_forward_steps: 204399
+forward_steps_in_memory: 40
+allow_incompatible_dataset: true
+checkpoint_path: ./ckpt.tar
+logging:
+  log_to_screen: true
+  log_to_wandb: false
+forcing_loader:
+  dataset:
+    data_path: ./data
+    file_pattern: 0211-0350.CM4_1pct_forcing_6hourly.zarr
+    engine: zarr
+  num_data_workers: 1
+initial_condition:
+  path: ./data/CM4_1pct_initial_condition.nc
+data_writer:
+  save_prediction_files: false
+  save_monthly_files: true

inference_config_OM4.yaml

@@ -0,0 +1,19 @@
+experiment_dir: ./inference/FloeNet_OM4_forcing
+n_forward_steps: 78892
+forward_steps_in_memory: 40
+allow_incompatible_dataset: true
+checkpoint_path: ./ckpt.tar
+logging:
+  log_to_screen: true
+  log_to_wandb: false
+forcing_loader:
+  dataset:
+    data_path: ./data
+    file_pattern: 1969-2022.OM4_forcing_6hourly.zarr
+    engine: zarr
+  num_data_workers: 1
+initial_condition:
+  path: ./data/OM4_initial_condition.nc
+data_writer:
+  save_prediction_files: false
+  save_monthly_files: true

inference_config_piControl.yaml

@@ -0,0 +1,19 @@
+experiment_dir: ./inference/FloeNet_piControl_forcing
+n_forward_steps: 204399
+forward_steps_in_memory: 40
+allow_incompatible_dataset: true
+checkpoint_path: ./ckpt.tar
+logging:
+  log_to_screen: true
+  log_to_wandb: false
+forcing_loader:
+  dataset:
+    data_path: ./data
+    file_pattern: 0211-0350.CM4_piControl_forcing_6hourly.zarr
+    engine: zarr
+  num_data_workers: 1
+initial_condition:
+  path: ./data/CM4_piControl_initial_condition.nc
+data_writer:
+  save_prediction_files: false
+  save_monthly_files: true