Add checkpoint and update README.md

.gitattributes

@@ -33,4 +33,3 @@ 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

@@ -1,29 +1,3 @@
----
-license: apache-2.0
----
-
-<img src="ACE-logo.png" alt="Logo for the ACE Project" style="width: auto; height: 50px;">
-
-# ACE2-SOM
-
-Ai2 Climate Emulator (ACE) is a family of models designed to simulate atmospheric variability from the time scale of days to centuries.
-
-**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 model, 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](https://arxiv.org/abs/2412.04418)
-- 💻 [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 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.  Prescribed sea-ice in particular means projections with ACE2-SOM are missing an important feedback mechanism known to amplify warming in the polar regions.
-- Abrupt regime shifts in stratospheric temperature and moisture can occur in inference runs with changing CO2 concentration.
-- Adjustment to an abrupt change in CO2 concentration occurs too quickly, violating energy conservation.
-- Sensitivities of the surface and top of atmosphere radiative fluxes to changes in CO2 with the atmospheric state held fixed are not always realistic.
+---
+license: apache-2.0
+---

ace2_som_ckpt.tar

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

forcing_data/1xCO2/forcing_2032.nc

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

forcing_data/1xCO2/forcing_2033.nc

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

forcing_data/1xCO2/forcing_2034.nc

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

forcing_data/1xCO2/forcing_2035.nc

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

forcing_data/3xCO2/forcing_2032.nc

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

forcing_data/3xCO2/forcing_2033.nc

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

forcing_data/3xCO2/forcing_2034.nc

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

forcing_data/3xCO2/forcing_2035.nc

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

inference_config.yaml

@@ -1,23 +0,0 @@
-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

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

initial_conditions/3xCO2/ic.nc

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