deepmd.pt.utils.learning_rate
=============================

.. py:module:: deepmd.pt.utils.learning_rate


Classes
-------

.. autoapisummary::

   deepmd.pt.utils.learning_rate.LearningRateExp


Module Contents
---------------

.. py:class:: LearningRateExp(start_lr: float, stop_lr: float, decay_steps: int, stop_steps: int, decay_rate: Optional[float] = None, **kwargs: Any)

   .. py:attribute:: start_lr


   .. py:attribute:: decay_steps


   .. py:attribute:: decay_rate


   .. py:attribute:: min_lr


   .. py:method:: value(step: int) -> numpy.float64

      
      Get the learning rate at the given step.
















      ..
          !! processed by numpydoc !!