tfm.optimization.DirectPowerDecay

Learning rate schedule follows lr * (step)^power.

initial_learning_rate The initial learning rate.
power The order of the polynomial.
name Optional, name of learning rate schedule.

Methods

from_config

Instantiates a LearningRateSchedule from its config.

Args
config Output of get_config().

Returns
A LearningRateSchedule instance.

get_config

View source

Get the configuration of the learning rate schedule.

__call__

View source

Call self as a function.