deepmd.pt.utils.learning_rate#

Classes#

Module Contents#

class deepmd.pt.utils.learning_rate.LearningRateExp(start_lr: float, stop_lr: float, decay_steps: int, stop_steps: int, decay_rate: float | None = None, **kwargs: Any)[source]#
start_lr#
decay_steps#
decay_rate#
min_lr#
value(step: int) numpy.float64[source]#

Get the learning rate at the given step.