InternalOptimizerGradientDescent.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "InternalOptimizerBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CInternalOptimizerGradientDescent : public CInternalOptimizerBase<T>
        {
        public:
            CInternalOptimizerGradientDescent();
            CInternalOptimizerGradientDescent(const CInternalOptimizerGradientDescent<T>& ob);
 
            virtual ~CInternalOptimizerGradientDescent();
 
            virtual const CResult Assign(const CInternalOptimizerGradientDescent<T>& ob);
            virtual const CResult Assign(const CInternalOptimizerGradientDescent<T>* pOb);
 
            virtual CInternalOptimizerBase<T>* Clone() const override;
 
            virtual const CResult Initialize() override;
 
            virtual const CResult Fit(T* pClippingThreshold = nullptr, int32_t* pI32IterationFeedbackFeedback = nullptr) override;
 
            virtual const CResult SetWeightDecay(T tWeightDecay);
            virtual T GetWeightDecay() const;
 
            virtual const CResult SetMaximized(bool bMaximized);
            virtual bool GetMaximized() const;
 
            virtual const CResult EnableSkip1DWeightDecay(bool bSkip1DWeightDecay);
            virtual bool IsSkip1DWeightDecayEnabled() const;
 
            virtual const CResult UpdateWeights() override;
 
 
            DeclareGetClassType();
 
        protected:
            T m_tWeightDecay;
            bool m_bMaximized;
            bool m_bSkip1DWeightDecay;
        };
 
        typedef CInternalOptimizerGradientDescent<float> CInternalOptimizerGradientDescentF;
        typedef CInternalOptimizerGradientDescent<double> CInternalOptimizerGradientDescentD;
 
        typedef CInternalOptimizerGradientDescent<float> CInternalOptimizerGDF;
        typedef CInternalOptimizerGradientDescent<double> CInternalOptimizerGDD;
 
        template <typename T>
        using CInternalOptimizerGD = CInternalOptimizerGradientDescent<T>;
    }
}
 
#endif