LearningRateSchedulerCosineAnnealing.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "LearningRateSchedulerBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CLearningRateSchedulerCosineAnnealing : public CLearningRateSchedulerBase<T>
        {
        public:
            CLearningRateSchedulerCosineAnnealing();
            CLearningRateSchedulerCosineAnnealing(const CLearningRateSchedulerCosineAnnealing<T>& lrs);
            CLearningRateSchedulerCosineAnnealing(const CLearningRateSchedulerCosineAnnealing<T>* pLrs);
 
            virtual ~CLearningRateSchedulerCosineAnnealing();
 
            virtual const CResult Assign(const CLearningRateSchedulerCosineAnnealing<T>& lrs);
            virtual const CResult Assign(const CLearningRateSchedulerCosineAnnealing<T>* pLrs);
 
            virtual const CResult Clear();
 
            virtual const CResult SetHalfCycle(int64_t i64HalfCycle = 50);
            virtual int64_t GetHalfCycle() const;
 
            virtual const CResult SetMinLearningRate(T tMinLearningRate = 0);
            virtual T GetMinLearningRate() const;
 
            virtual const CResult Step(T tLoss = 0) override;
 
 
            DeclareGetClassType();
            SupportToDuplicateObject(CLearningRateSchedulerCosineAnnealing<T>, *this);
 
        protected:
            int64_t m_i64HalfCycle;
            T m_tMinLearningRate;
        };
 
 
        typedef CLearningRateSchedulerCosineAnnealing<float> CLearningRateSchedulerCosineAnnealingF;
        typedef CLearningRateSchedulerCosineAnnealing<double> CLearningRateSchedulerCosineAnnealingD;
 
        typedef CLearningRateSchedulerCosineAnnealing<float> CLRSCosineAnnealingF;
        typedef CLearningRateSchedulerCosineAnnealing<double> CLRSCosineAnnealingD;
 
        template <typename T>
        using CLRSCosineAnnealing = CLearningRateSchedulerCosineAnnealing<T>;
 
    }
}
 
#endif