Regression in segmentation training

Incorrect copying of hyperparameter data. Fixes #600.
mittagessen · May 5, 2024 · 2eba59f · 2eba59f
1 parent 5fe2972
commit 2eba59f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kraken/lib/train.py b/kraken/lib/train.py
@@ -1016,7 +1016,7 @@ def setup(self, stage: Optional[str] = None):
                 self.val_set.dataset.class_mapping = self.nn.user_metadata['class_mapping']
 
             # updates model's hyper params with user-defined ones
-            self.nn.hyper_params = self.hparams
+            self.nn.hyper_params = self.hparams.hyper_params
 
             # change topline/baseline switch
             loc = {None: 'centerline',