optimiser_setup_base Module Function

private module function optimiser_setup_base(learning_rate, num_params, regulariser, clip_dict, lr_decay) result(optimiser)

Set up the base optimiser

Arguments

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

Learning rate
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(base_optimiser_type)

Instance of the base optimiser


Source Code

     module function optimiser_setup_base( &
          learning_rate, num_params, &
          regulariser, clip_dict, lr_decay) result(optimiser)
       !! Set up the base optimiser
       real(real32), optional, intent(in) :: learning_rate
       !! Learning rate
       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(base_optimiser_type) :: optimiser
       !! Instance of the base optimiser
     end function optimiser_setup_base