From 6bad1d2bf69ead158ee44c21cd4d2528b6569007 Mon Sep 17 00:00:00 2001 From: Colin Brisson Date: Wed, 22 Feb 2023 21:06:15 +0000 Subject: [PATCH] update --warmup option help text --- kraken/ketos/recognition.py | 2 +- kraken/ketos/segmentation.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/kraken/ketos/recognition.py b/kraken/ketos/recognition.py index 8611ba5d1..7cfae5ca7 100644 --- a/kraken/ketos/recognition.py +++ b/kraken/ketos/recognition.py @@ -88,7 +88,7 @@ @click.option('-w', '--weight-decay', show_default=True, type=float, default=RECOGNITION_HYPER_PARAMS['weight_decay'], help='Weight decay') @click.option('--warmup', show_default=True, type=int, - default=RECOGNITION_HYPER_PARAMS['warmup'], help='Number of samples to ramp up to `lrate` initial learning rate.') + default=RECOGNITION_HYPER_PARAMS['warmup'], help='Number of steps to ramp up to `lrate` initial learning rate.') @click.option('--freeze-backbone', show_default=True, type=int, default=RECOGNITION_HYPER_PARAMS['freeze_backbone'], help='Number of samples to keep the backbone (everything but last layer) frozen.') @click.option('--schedule', diff --git a/kraken/ketos/segmentation.py b/kraken/ketos/segmentation.py index 1834a5055..6b5f9f259 100644 --- a/kraken/ketos/segmentation.py +++ b/kraken/ketos/segmentation.py @@ -112,7 +112,7 @@ def _validate_merging(ctx, param, value): @click.option('-w', '--weight-decay', show_default=True, default=SEGMENTATION_HYPER_PARAMS['weight_decay'], help='Weight decay') @click.option('--warmup', show_default=True, type=float, - default=SEGMENTATION_HYPER_PARAMS['warmup'], help='Number of samples to ramp up to `lrate` initial learning rate.') + default=SEGMENTATION_HYPER_PARAMS['warmup'], help='Number of steps to ramp up to `lrate` initial learning rate.') @click.option('--schedule', show_default=True, type=click.Choice(['constant',