mediapipe_model_maker.model_util.WarmUp

Applies a warmup schedule on a given learning rate decay schedule.

Main aliases

mediapipe_model_maker.face_stylizer.face_stylizer.face_stylizer_options.model_opt.loss_functions.model_util.WarmUp, mediapipe_model_maker.face_stylizer.face_stylizer.loss_functions.model_util.WarmUp, mediapipe_model_maker.face_stylizer.face_stylizer.model_opt.loss_functions.model_util.WarmUp, mediapipe_model_maker.face_stylizer.face_stylizer.model_util.WarmUp, mediapipe_model_maker.face_stylizer.face_stylizer_options.model_opt.loss_functions.model_util.WarmUp, mediapipe_model_maker.face_stylizer.model_options.loss_functions.model_util.WarmUp

initial_learning_rate learning rate after the warmup.
decay_schedule_fn A function maps step to learning rate. Will be applied for values of step larger than 'warmup_steps'.
warmup_steps Number of steps to do warmup for.
name TF namescope under which to perform the learning rate calculation.

Methods

from_config

Instantiates a LearningRateSchedule from its config.

Args
config Output of get_config().

Returns
A LearningRateSchedule instance.

get_config

View source

__call__

View source

Call self as a function.