rename configs to match dataset description

rather than library used to read data i.e. xarray -> ukcp18
henryaddison · Feb 9, 2024 · ab192ec · ab192ec
1 parent 2a5103e
commit ab192ec
Show file tree

Hide file tree

Showing 11 changed files with 7 additions and 194 deletions.
diff --git a/bin/local-test-train b/bin/local-test-train
@@ -2,7 +2,7 @@
 
 set -euo pipefail
 
-config_name="xarray_cunet_continuous"
+config_name="ukcp18_cunet_continuous"
 dataset="bham_gcmx-4x_12em_psl-temp4th-vort4th_eqvt_random-season"
 
 sde="subvpsde"

diff --git a/...22/configs/default_xarray_12em_configs.py → ...22/configs/default_ukcp18_12em_configs.py b/...22/configs/default_xarray_12em_configs.py → ...22/configs/default_ukcp18_12em_configs.py
@@ -1,7 +1,7 @@
 import ml_collections
 import torch
 
-from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_xarray_configs import get_default_configs as get_base_configs
+from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_ukcp18_configs import get_default_configs as get_base_configs
 
 
 def get_default_configs():

diff --git a/...h_hja22/configs/default_xarray_configs.py → ...h_hja22/configs/default_ukcp18_configs.py b/...h_hja22/configs/default_xarray_configs.py → ...h_hja22/configs/default_ukcp18_configs.py
diff --git a/...ubvpsde/xarray_12em_cncsnpp_continuous.py → ...ubvpsde/ukcp18_12em_cncsnpp_continuous.py b/...ubvpsde/xarray_12em_cncsnpp_continuous.py → ...ubvpsde/ukcp18_12em_cncsnpp_continuous.py
@@ -15,7 +15,7 @@
 
 # Lint as: python3
 """Training NCSN++ on precip data with sub-VP SDE."""
-from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_xarray_12em_configs import get_default_configs
+from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_ukcp18_12em_configs import get_default_configs
 
 
 def get_config():

diff --git a/...igs/subvpsde/xarray_cncsnpp_continuous.py → ...igs/subvpsde/ukcp18_cncsnpp_continuous.py b/...igs/subvpsde/xarray_cncsnpp_continuous.py → ...igs/subvpsde/ukcp18_cncsnpp_continuous.py
@@ -15,7 +15,7 @@
 
 # Lint as: python3
 """Training NCSN++ on precip data with sub-VP SDE."""
-from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_xarray_configs import get_default_configs
+from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_ukcp18_configs import get_default_configs
 
 
 def get_config():

diff --git a/...nfigs/subvpsde/xarray_cunet_continuous.py → ...nfigs/subvpsde/ukcp18_cunet_continuous.py b/...nfigs/subvpsde/xarray_cunet_continuous.py → ...nfigs/subvpsde/ukcp18_cunet_continuous.py
@@ -15,7 +15,7 @@
 
 # Lint as: python3
 """Training conditional U-Net on precip data with sub-VP SDE."""
-from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_xarray_configs import get_default_configs
+from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_ukcp18_configs import get_default_configs
 
 
 def get_config():

diff --git a/...downscaling_emulator/score_sde_pytorch_hja22/configs/subvpsde/xarray_ncsnpp_continuous.py b/...downscaling_emulator/score_sde_pytorch_hja22/configs/subvpsde/xarray_ncsnpp_continuous.py
diff --git a/...onfigs/vesde/xarray_cncsnpp_continuous.py → ...onfigs/vesde/ukcp18_cncsnpp_continuous.py b/...onfigs/vesde/xarray_cncsnpp_continuous.py → ...onfigs/vesde/ukcp18_cncsnpp_continuous.py
@@ -15,7 +15,7 @@
 
 # Lint as: python3
 """Training NCSN++ on precip data with VE SDE."""
-from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_xarray_configs import get_default_configs
+from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_ukcp18_configs import get_default_configs
 
 
 def get_config():

diff --git a/.../configs/vesde/xarray_cunet_continuous.py → .../configs/vesde/ukcp18_cunet_continuous.py b/.../configs/vesde/xarray_cunet_continuous.py → .../configs/vesde/ukcp18_cunet_continuous.py
@@ -15,7 +15,7 @@
 
 # Lint as: python3
 """Training UNet on XArray with VE SDE."""
-from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_xarray_configs import get_default_configs
+from ml_downscaling_emulator.score_sde_pytorch_hja22.configs.default_ukcp18_configs import get_default_configs
 
 
 def get_config():

diff --git a/...ml_downscaling_emulator/score_sde_pytorch_hja22/configs/vesde/xarray_ncsnpp_continuous.py b/...ml_downscaling_emulator/score_sde_pytorch_hja22/configs/vesde/xarray_ncsnpp_continuous.py
diff --git a/src/ml_downscaling_emulator/score_sde_pytorch_hja22/configs/vesde/xarray_unet_continuous.py b/src/ml_downscaling_emulator/score_sde_pytorch_hja22/configs/vesde/xarray_unet_continuous.py