optimiser_setup_rmsprop Module Function

private module function optimiser_setup_rmsprop(learning_rate, beta, epsilon, num_params, regulariser, clip_dict, lr_decay) result(optimiser)

Set up the RMSprop optimiser

Arguments

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

Learning rate
real(kind=real32), intent(in), optional :: beta

Beta and epsilon
real(kind=real32), intent(in), optional :: epsilon

Beta and epsilon
integer, intent(in), optional :: num_params

Number of parameters
class(base_regulariser_type), intent(in), optional :: regulariser

Regularisation method
type(clip_type), intent(in), optional :: clip_dict

Clipping dictionary
class(base_lr_decay_type), intent(in), optional :: lr_decay

Learning rate decay method

Return Value type(rmsprop_optimiser_type)

Instance of the RMSprop optimiser


Source Code

     module function optimiser_setup_rmsprop( &
          learning_rate, beta, &
          epsilon, num_params, &
          regulariser, clip_dict, lr_decay) result(optimiser)
       !! Set up the RMSprop optimiser
       real(real32), optional, intent(in) :: learning_rate, beta, epsilon
       !! Learning rate, beta, and epsilon
       integer, optional, intent(in) :: num_params
       !! Number of parameters
       class(base_regulariser_type), optional, intent(in) :: regulariser
       !! Regularisation method
       type(clip_type), optional, intent(in) :: clip_dict
       !! Clipping dictionary
       class(base_lr_decay_type), optional, intent(in) :: lr_decay
       !! Learning rate decay method
       type(rmsprop_optimiser_type) :: optimiser
       !! Instance of the RMSprop optimiser
     end function optimiser_setup_rmsprop