corrected test super config + got rid of deprecated val_split_super r…

…eference
RolnickLab · Apr 1, 2024 · c2b96e2 · c2b96e2
1 parent cf5d5fa
commit c2b96e2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/emulator/configs/experiment/test_super.yaml b/emulator/configs/experiment/test_super.yaml
@@ -33,7 +33,7 @@ trainer:
 model:
   loss_function: "climax_lon_lat_rmse"
   monitor: "val/llrmse_climax"
-  finetune: True
+  finetune: False
   pretrained_run_id:  "bcvgpg4b"
   pretrained_ckpt_dir: "home/causalpaca/emulator/emulator/bcvgpg4b/checkpoints/epoch=8-step=180.ckpt"
   pretrained_run_id_decoder: "xog84o5g"
@@ -55,9 +55,6 @@ datamodule: # overwrite what stuff to train on
   test_scenarios : ["ssp370"]
   train_models : ["MPI-ESM1-2-HR", "GFDL-ESM4", "NorESM2-LM"]
   test_models: ["NorESM2-LM"]
-  #num_ensembles: 2
-  output_save_dir : '/home/mila/c/charlotte.lange/scratch/causalpaca/emulator/DATA'
-  #cmip6_data_dir: '/home/mila/v/venkatesh.ramesh/scratch/causal_data/dl_fromcc/'
 
 logger:
   wandb:

diff --git a/emulator/src/datamodules/super_climate_datamodule.py b/emulator/src/datamodules/super_climate_datamodule.py
@@ -15,11 +15,10 @@
     NUM_LEVELS,
     DATA_DIR,
 )
-from emulator.src.utils.utils import get_logger, random_split, random_split_super
+from emulator.src.utils.utils import get_logger
 
 log = get_logger()
 
-
 class SuperClimateDataModule(LightningDataModule):
     """
     ----------------------------------------------------------------------------------------------------------