update --warmup option help text

mittagessen · Feb 22, 2023 · 6bad1d2 · 6bad1d2
1 parent 8e297b4
commit 6bad1d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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',