Add initial condition and forcing data for 1xCO2 and 3xCO2

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.nc filter=lfs diff=lfs merge=lfs -text

README.md

@@ -10,19 +10,17 @@ Ai2 Climate Emulator (ACE) is a family of models designed to simulate atmospheri
 
 **Disclaimer: ACE models are research tools and should not be used for operational climate predictions.**
 
-ACE2-SOM is the first model we have trained to emulate the sensitivity of climate to substantial increases in carbon dioxide.  It consists of ACE2 coupled to a highly simplified physics-based ocean mode, called a slab ocean model, and is trained on equilibrium-climate output from GFDL's SHiELD model coupled to an identical slab ocean with carbon dioxide concentrations of the present-day, double the present-day, and quadruple the present-day.
+ACE2-SOM is the first model we have trained to emulate the sensitivity of climate to substantial increases in carbon dioxide.  It consists of ACE2 coupled to a highly simplified physics-based ocean mode, called a slab ocean model (SOM), and is trained on equilibrium-climate output from GFDL's SHiELD model coupled to an identical slab ocean with carbon dioxide concentrations of the present-day, double the present-day, and quadruple the present-day (1xCO2, 2xCO2, and 4xCO2).
 
 Quick links:
-- 📃 [Paper (TODO: update)]()
+- 📃 [Paper (pre-print coming soon)]()
 - 💻 [Code](https://github.com/ai2cm/ace)
 - 💬 [Docs](https://ai2-climate-emulator.readthedocs.io/en/stable/)
 - 📂 [All Models](https://huggingface.co/collections/allenai/ace-67327d822f0f0d8e0e5e6ca4)
 
 Briefly, the strengths of ACE2-SOM are:
-- accurate emulation equilibrium climates with CO2 concentrations ranging from the present-day to four times the present-day concentration, including
-  - the spatial pattern of temperature and precipitation changes between climates
-  - changes in precipitation extremes between climates
-- realistic emulation non-stratospheric fields with gradually changing CO2 concentration between the present-day and four times the present-day concentration.
+- accurate emulation of equilibrium climates with CO2 concentrations ranging from 1xCO2 to 4xCO2, including changes in the spatial patterns of temperature and precipitation, and changes in precipitation extremes between climates
+- realistic emulation of surface and lower-to-middle atmosphere fields with gradually changing CO2 concentration between 1xCO2 and 4xCO2.
 
 Some known weaknesses are:
 - Sea-ice coverage, ocean heat transport, and ocean mixed layer depth are prescribed based on a present-day climatologies and therefore do not respond to changes in CO2.  Presribed sea-ice in particular means projections with ACE2-SOM are missing an important feedback mechanism known to amplify warming in the polar regions.

forcing_data/1xCO2/forcing_2032.nc

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

forcing_data/1xCO2/forcing_2033.nc

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

forcing_data/1xCO2/forcing_2034.nc

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

forcing_data/1xCO2/forcing_2035.nc

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

forcing_data/3xCO2/forcing_2032.nc

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

forcing_data/3xCO2/forcing_2033.nc

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

forcing_data/3xCO2/forcing_2034.nc

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

forcing_data/3xCO2/forcing_2035.nc

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

inference_config.yaml

@@ -0,0 +1,23 @@
+experiment_dir: /output_directory
+n_forward_steps: 400 
+forward_steps_in_memory: 50
+checkpoint_path: /ace2_som_ckpt.tar
+logging:
+  log_to_screen: true
+  log_to_wandb: false
+  log_to_file: true
+  project: ace
+initial_condition:
+  path: /initial_conditions/1xCO2/ic.nc
+  start_indices:
+    times:
+      - "2032-01-01T00:00:00"
+forcing_loader:
+  dataset:
+    data_path: /forcing_data/1xCO2
+    n_repeats: 1  # Use this to extend the four years of forcing data to the desired length
+  num_data_workers: 4
+data_writer:
+  save_prediction_files: true
+  save_monthly_files: false
+  names: ["TMP2m", "VGRD10m", "PRATEsfc"]

initial_conditions/1xCO2/ic.nc

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

initial_conditions/3xCO2/ic.nc

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