deepmd.pt.utils.learning_rate

Module Contents

Classes

LearningRateExp

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

Get the learning rate at the given step.