Removed duplicated n_epoch parameters

CosmoStat · Oct 31, 2023 · f94afc0 · f94afc0
1 parent 36547c2
commit f94afc0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/config/training_config.yaml b/config/training_config.yaml
@@ -88,11 +88,6 @@ training:
 
   # Training hyperparameters
   training_hparams:
-    # Number of epochs for the training the parametric model parameters
-    n_epochs_params: [2, 2]
-
-    # Number of epochs for the training the non-parametric model parameters
-    n_epochs_non_params: [2, 2] 
 
     # Batch Size
     batch_size: 32

diff --git a/src/wf_psf/training/train.py b/src/wf_psf/training/train.py
@@ -147,7 +147,7 @@ def total_cycles(self):
 
     @property
     def n_epochs_params(self):
-        """Number of Epochs for Pparametric PSF model.
+        """Number of Epochs for Parametric PSF model.
 
         Set the number of epochs for
         training parametric PSF model.
@@ -158,7 +158,7 @@ def n_epochs_params(self):
             List of number of epochs for training parametric PSF model.
 
         """
-        return self.training_hparams.n_epochs_params
+        return self.multi_cycle_params.n_epochs_params
 
     @property
     def n_epochs_non_params(self):
@@ -173,7 +173,7 @@ def n_epochs_non_params(self):
             List of number of epochs for training non-parametric PSF model.
 
         """
-        return self.training_hparams.n_epochs_non_params
+        return self.multi_cycle_params.n_epochs_non_params
 
     @property
     def learning_rate_params(self):