LearningRateSchedulerMultiStep.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "LearningRateSchedulerBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CLearningRateSchedulerMultiStep : public CLearningRateSchedulerBase<T>
        {
        public:
            enum EMilestoneMode
            {
                EMilestoneMode_Unit = 0,
                EMilestoneMode_Ratio,
            };
 
            CLearningRateSchedulerMultiStep();
            CLearningRateSchedulerMultiStep(const CLearningRateSchedulerMultiStep<T>& lrs);
            CLearningRateSchedulerMultiStep(const CLearningRateSchedulerMultiStep<T>* pLrs);
 
            virtual ~CLearningRateSchedulerMultiStep();
 
            virtual const CResult Assign(const CLearningRateSchedulerMultiStep<T>& lrs);
            virtual const CResult Assign(const CLearningRateSchedulerMultiStep<T>* pLrs);
 
            virtual const CResult Clear();
            virtual const CResult Initialize() override;
 
            virtual const CResult SetMilestoneMode(EMilestoneMode eMilestoneMode = EMilestoneMode_Unit);
            virtual EMilestoneMode GetMilestoneMode() const;
 
            virtual const CResult SetMilestones(const std::vector<T>& vctMilestones);
            virtual const std::vector<T>& GetMilestones() const;
 
            virtual const CResult SetGamma(T tGamma = (T)0.5);
            virtual T GetGamma() const;
 
            virtual const CResult Step(T tLoss = 0) override;
 
            virtual const CResult ConvertMilestoneToMilestoneUnit(int64_t i64TotalCount = 0);
 
 
            DeclareGetClassType();
            SupportToDuplicateObject(CLearningRateSchedulerMultiStep<T>, *this);
 
        protected:
            EMilestoneMode m_eMilestoneMode;
            std::vector<T>& m_vctMilestones;
            std::vector<int64_t>& m_vctActualMilestonesUnit;
            T m_tGamma;
            int64_t m_i64CurrentIndex;
        };
 
 
        typedef CLearningRateSchedulerMultiStep<float> CLearningRateSchedulerMultiStepF;
        typedef CLearningRateSchedulerMultiStep<double> CLearningRateSchedulerMultiStepD;
 
        typedef CLearningRateSchedulerMultiStep<float> CLRSMultiStepF;
        typedef CLearningRateSchedulerMultiStep<double> CLRSMultiStepD;
 
        template <typename T>
        using CLRSMultiStep = CLearningRateSchedulerMultiStep<T>;
 
    }
}
 
#endif