Deep Learning Algorithm Implementations 1.0.0
C++ implementations of fundamental deep learning algorithms
Loading...
Searching...
No Matches
dl::optimization::LRScheduler< T > Class Template Referenceabstract

Learning rate scheduler base class. More...

#include <optimizers.hpp>

Inheritance diagram for dl::optimization::LRScheduler< T >:

Public Member Functions

 LRScheduler (AutogradOptimizer< T > *optimizer)
 
virtual ~LRScheduler ()=default
 
virtual void step ()=0
 Update learning rate.
 
get_lr () const
 Get current learning rate.
 

Protected Attributes

AutogradOptimizer< T > * optimizer_
 

Detailed Description

template<typename T>
class dl::optimization::LRScheduler< T >

Learning rate scheduler base class.

Definition at line 295 of file optimizers.hpp.

Constructor & Destructor Documentation

◆ LRScheduler()

template<typename T >
dl::optimization::LRScheduler< T >::LRScheduler ( AutogradOptimizer< T > *  optimizer)
inlineexplicit

Definition at line 297 of file optimizers.hpp.

◆ ~LRScheduler()

template<typename T >
virtual dl::optimization::LRScheduler< T >::~LRScheduler ( )
virtualdefault

Member Function Documentation

◆ get_lr()

template<typename T >
T dl::optimization::LRScheduler< T >::get_lr ( ) const
inline

Get current learning rate.

Definition at line 308 of file optimizers.hpp.

◆ step()

template<typename T >
virtual void dl::optimization::LRScheduler< T >::step ( )
pure virtual

Update learning rate.

Implemented in dl::optimization::StepLR< T >.

Member Data Documentation

◆ optimizer_

template<typename T >
AutogradOptimizer<T>* dl::optimization::LRScheduler< T >::optimizer_
protected

Definition at line 311 of file optimizers.hpp.


The documentation for this class was generated from the following file: