LearningRateSchedulerStep.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "LearningRateSchedulerBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CLearningRateSchedulerStep : public CLearningRateSchedulerBase<T>
        {
        public:
            CLearningRateSchedulerStep();
            CLearningRateSchedulerStep(const CLearningRateSchedulerStep<T>& lrs);
            CLearningRateSchedulerStep(const CLearningRateSchedulerStep<T>* pLrs);
 
            virtual ~CLearningRateSchedulerStep();
 
            virtual const CResult Assign(const CLearningRateSchedulerStep<T>& lrs);
            virtual const CResult Assign(const CLearningRateSchedulerStep<T>* pLrs);
 
            virtual const CResult Clear();
 
            virtual const CResult SetReductionUnit(int64_t i64ReductionUnit = 10);
            virtual int64_t GetReductionUnit() const;
 
            virtual const CResult SetGamma(T tGamma = (T)0.5);
            virtual T GetGamma() const;
 
            virtual const CResult Step(T tLoss = 0) override;
 
 
            DeclareGetClassType();
            SupportToDuplicateObject(CLearningRateSchedulerStep<T>, *this);
 
        protected:
            int64_t m_i64ReductionUnit;
            T m_tGamma;
        };
 
 
        typedef CLearningRateSchedulerStep<float> CLearningRateSchedulerStepF;
        typedef CLearningRateSchedulerStep<double> CLearningRateSchedulerStepD;
 
        typedef CLearningRateSchedulerStep<float> CLRSStepF;
        typedef CLearningRateSchedulerStep<double> CLRSStepD;
 
        template <typename T>
        using CLRSStep = CLearningRateSchedulerStep<T>;
 
    }
}
 
#endif