InternalOptimizerLionGradientDescent.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "InternalOptimizerMomentumGradientDescent.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CInternalOptimizerLionGradientDescent : public CInternalOptimizerMomentumGradientDescent<T>
        {
        public:
            CInternalOptimizerLionGradientDescent();
            CInternalOptimizerLionGradientDescent(const CInternalOptimizerLionGradientDescent<T>& ob);
 
            virtual ~CInternalOptimizerLionGradientDescent();
 
            virtual const CResult Assign(const CInternalOptimizerLionGradientDescent<T>& ob);
            virtual const CResult Assign(const CInternalOptimizerLionGradientDescent<T>* pOb);
 
            virtual CInternalOptimizerBase<T>* Clone() const override;
 
            virtual const CResult Initialize() override;
 
            virtual const CResult SetBeta1(T tBeta1);
            virtual T GetBeta1() const;
 
            virtual const CResult SetBeta2(T tBeta2);
            virtual T GetBeta2() const;
 
            virtual const CResult UpdateWeights() override;
 
 
            DeclareGetClassType();
 
        protected:
 
            T& m_tBeta1;
            T m_tBeta2;
            
        private:
 
            using CInternalOptimizerMomentumGradientDescent<T>::GetAlpha;
            using CInternalOptimizerMomentumGradientDescent<T>::SetAlpha;
        };
 
 
        typedef CInternalOptimizerLionGradientDescent<float> CInternalOptimizerLionGradientDescentF;
        typedef CInternalOptimizerLionGradientDescent<double> CInternalOptimizerLionGradientDescentD;
 
        typedef CInternalOptimizerLionGradientDescent<float> CInternalOptimizerLionGDF;
        typedef CInternalOptimizerLionGradientDescent<double> CInternalOptimizerLionGDD;
 
        template <typename T>
        using CInternalOptimizerLionGD = CInternalOptimizerLionGradientDescent<T>;
    }
}
 
#endif