Create a new learning rate schedule type
new_learning_rate_schedule_class(
classname,
...,
initialize = NULL,
call,
get_config = NULL
)
A LearningRateSchedule
class generator.
string
methods and properties of the schedule class
Additional recommended methods to implement.
function which takes a step argument (scalar integer tensor, the
current training step count, and returns the new learning rate). For
tracking additional state, objects self
and private
are automatically
injected into the scope of the function.