LearningRateSchedulerLambda.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "LearningRateSchedulerBase.h"
 
#include <functional>
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CLearningRateSchedulerLambda : public CLearningRateSchedulerBase<T>
        {
        public:
            enum EOperationMode
            {
                EOperationMode_Normal = 0,
                EOperationMode_Accumulate
            };
 
            CLearningRateSchedulerLambda();
            CLearningRateSchedulerLambda(const CLearningRateSchedulerLambda<T>& lrs);
            CLearningRateSchedulerLambda(const CLearningRateSchedulerLambda<T>* pLrs);
 
            virtual ~CLearningRateSchedulerLambda();
 
            virtual const CResult Assign(const CLearningRateSchedulerLambda<T>& lrs);
            virtual const CResult Assign(const CLearningRateSchedulerLambda<T>* pLrs);
 
            virtual const CResult Clear();
            virtual const CResult Initialize() override;
 
            virtual const CResult SetLambda(std::function<T(T, int64_t)>& funcLambda);
            virtual const std::function<T(T, int64_t)>* GetLambda() const;
 
            virtual const CResult SetOperationMode(EOperationMode eOperationMode = EOperationMode_Normal);
            virtual EOperationMode GetOperationMode() const;
 
            virtual const CResult Step(T tLoss = 0) override;
 
 
            DeclareGetClassType();
            SupportToDuplicateObject(CLearningRateSchedulerLambda<T>, *this);
 
        protected:
            std::function<T(T, int64_t)>* m_pFuncLambda; // lr_new = func(lr_org, step)
            EOperationMode m_eOperationMode;
        };
 
 
        typedef CLearningRateSchedulerLambda<float> CLearningRateSchedulerLambdaF;
        typedef CLearningRateSchedulerLambda<double> CLearningRateSchedulerLambdaD;
 
        typedef CLearningRateSchedulerLambda<float> CLRSLambdaF;
        typedef CLearningRateSchedulerLambda<double> CLRSLambdaD;
 
        template <typename T>
        using CLRSLambda = CLearningRateSchedulerLambda<T>;
 
    }
}
 
#endif