Merge pull request #163 from openclimatefix/update-experiment-config-…

…examples Remove references to old datamodule parameters in config examples
openclimatefix · Mar 19, 2024 · 7299ea8 · 7299ea8
2 parents 27f16c5 + cc75c26
commit 7299ea8
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 11 deletions.
diff --git a/configs.example/experiment/baseline.yaml b/configs.example/experiment/baseline.yaml
@@ -19,7 +19,3 @@ validate_only: "1" # by putting this key in the config file, the model does not
 trainer:
   min_epochs: 1
   max_epochs: 1
-
-datamodule:
-  n_train_data: 2
-  n_val_data: 10
diff --git a/configs.example/experiment/conv3d_sat_nwp.yaml b/configs.example/experiment/conv3d_sat_nwp.yaml
@@ -19,9 +19,5 @@ trainer:
   min_epochs: 1
   max_epochs: 10
 
-datamodule:
-  n_train_data: 4000
-  n_val_data: 400
-
 model:
   conv3d_channels: 32
diff --git a/configs.example/experiment/example_simple.yaml b/configs.example/experiment/example_simple.yaml
@@ -22,8 +22,6 @@ trainer:
   max_epochs: 2
 
 datamodule:
-  n_train_data: 2
-  n_val_data: 2
-  fake_data: 1
+  batch_size: 32
 
 validate_only: "1" # by putting this key in the config file, the model does not get trained.