From db35eaef0c56c8750a16f8234e36371e3b0d1979 Mon Sep 17 00:00:00 2001 From: Benjamin Kiessling Date: Thu, 4 Apr 2024 23:41:12 +0200 Subject: [PATCH] fix hyper param dicts --- kraken/ketos/pretrain.py | 2 +- kraken/ketos/ro.py | 2 +- kraken/ketos/segmentation.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/kraken/ketos/pretrain.py b/kraken/ketos/pretrain.py index 27eabfd1f..841d39191 100644 --- 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 index a8b7e4c9b..055d86d29 100644 --- 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 index 171e834f1..e91e4e2cd 100644 --- 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')