deepmd.pt.utils.learning_rate#

Classes#

Module Contents#

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

Get the learning rate at the given step.