fix hyper param dicts

mittagessen · Apr 4, 2024 · db35eae · db35eae
1 parent f7fb622
commit db35eae
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/kraken/ketos/pretrain.py b/kraken/ketos/pretrain.py
@@ -125,7 +125,7 @@
               help='Epoch of minimal learning rate for cosine LR scheduler.')
 @click.option('--cos-min-lr',
               show_default=True,
-              default=RECOGNITION_HYPER_PARAMS['cos_min_lr'],
+              default=RECOGNITION_PRETRAIN_HYPER_PARAMS['cos_min_lr'],
               help='Minimal final learning rate for cosine LR scheduler.')
 @click.option('-p', '--partition', show_default=True, default=0.9,
               help='Ground truth data partition ratio between train/validation set')

diff --git a/kraken/ketos/ro.py b/kraken/ketos/ro.py
@@ -117,7 +117,7 @@
               help='Epoch of minimal learning rate for cosine LR scheduler.')
 @click.option('--cos-min-lr',
               show_default=True,
-              default=RECOGNITION_HYPER_PARAMS['cos_min_lr'],
+              default=READING_ORDER_HYPER_PARAMS['cos_min_lr'],
               help='Minimal final learning rate for cosine LR scheduler.')
 @click.option('-p', '--partition', show_default=True, default=0.9,
               help='Ground truth data partition ratio between train/validation set')

diff --git a/kraken/ketos/segmentation.py b/kraken/ketos/segmentation.py
@@ -153,7 +153,7 @@ def _validate_merging(ctx, param, value):
               help='Epoch of minimal learning rate for cosine LR scheduler.')
 @click.option('--cos-min-lr',
               show_default=True,
-              default=RECOGNITION_HYPER_PARAMS['cos_min_lr'],
+              default=SEGMENTATION_HYPER_PARAMS,['cos_min_lr'],
               help='Minimal final learning rate for cosine LR scheduler.')
 @click.option('-p', '--partition', show_default=True, default=0.9,
               help='Ground truth data partition ratio between train/validation set')