setup_lr_decay_step Module Function

private module function setup_lr_decay_step(decay_rate, decay_steps) result(lr_decay)

Set up step learning rate decay type

Arguments

Type IntentOptional Attributes Name
real(kind=real32), intent(in), optional :: decay_rate

Decay rate for learning rate
integer, intent(in), optional :: decay_steps

Number of steps for learning rate decay

Return Value type(step_lr_decay_type)

Step learning rate decay type


Source Code

     module function setup_lr_decay_step(decay_rate, decay_steps) &
          result(lr_decay)
       !! Set up step learning rate decay type
       real(real32), optional, intent(in) :: decay_rate
       !! Decay rate for learning rate
       integer, optional, intent(in) :: decay_steps
       !! Number of steps for learning rate decay
       type(step_lr_decay_type) :: lr_decay
       !! Step learning rate decay type
     end function setup_lr_decay_step