Fix regression in segtrain

mittagessen · Feb 22, 2024 · e083808 · e083808
1 parent 4fdd985
commit e083808
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/kraken/lib/train.py b/kraken/lib/train.py
@@ -839,8 +839,7 @@ def __init__(self,
             train_set.add(page)
 
         if evaluation_data:
-            val_set = BaselineSet(evaluation_data,
-                                  line_width=self.hparams.hyper_params['line_width'],
+            val_set = BaselineSet(line_width=self.hparams.hyper_params['line_width'],
                                   im_transforms=transforms,
                                   augmentation=False,
                                   valid_baselines=valid_baselines,