Source code for otx.algorithms.segmentation.adapters.mmseg.models.schedulers.constant

"""Constant scheduler."""
# Copyright (C) 2023 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
#

from otx.algorithms.segmentation.adapters.mmseg.utils.builder import SCALAR_SCHEDULERS

from .base import BaseScalarScheduler




@SCALAR_SCHEDULERS.register_module()
class ConstantScalarScheduler(BaseScalarScheduler):
    """The learning rate remains constant over time.

    The learning rate equals the scale.

    Args:
        scale (float): The learning rate scale.
    """

    def __init__(self, scale: float = 30.0):
        super().__init__()

        self._end_s = scale
        assert self._end_s > 0.0

    def _get_value(self, step, epoch_size):
        return self._end_s