mediapipe_model_maker.model_util.WarmUp

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

      mediapipe_model_maker 
 . 
 model_util 
 . 
 WarmUp 
 ( 
 initial_learning_rate 
 : 
 float 
 , 
 decay_schedule_fn 
 : 
 Callable 
 [[ 
 Any 
 ], 
 Any 
 ], 
 warmup_steps 
 : 
 int 
 , 
 name 
 : 
 Optional 
 [ 
 str 
 ] 
 = 
 None 
 )

    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 

      @classmethod 
 
  from_config 
 ( 
 config 
 )

    Instantiates a LearningRateSchedule from its config.

    Args

    config
    Output of get_config() .

    Returns
    A LearningRateSchedule instance.

    get_config

    View source 

      get_config 
 () 
 -> 
 Dict 
 [ 
 str 
 , 
 Any 
 ]

    __call__

    View source 

      __call__ 
 ( 
 step 
 : 
 Union 
 [ 
 int 
 , 
 tf 
 . 
 Tensor 
 ] 
 ) 
 -> 
 tf 
 . 
 Tensor

    Call self as a function.
    Create a Mobile Website
    View Site in Mobile | Classic
    Share by: