InternalOptimizerSpecGradientDescent.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "InternalOptimizerSpecBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CInternalOptimizerSpecGradientDescent : public CInternalOptimizerSpecBase<T>
        {
        public:
            CInternalOptimizerSpecGradientDescent();
            CInternalOptimizerSpecGradientDescent(const CInternalOptimizerSpecGradientDescent<T>& osb);
            CInternalOptimizerSpecGradientDescent(const CInternalOptimizerSpecGradientDescent<T>* pOsb);
 
            virtual ~CInternalOptimizerSpecGradientDescent();
 
            virtual const CResult Assign(const CInternalOptimizerSpecGradientDescent<T>& osb);
            virtual const CResult Assign(const CInternalOptimizerSpecGradientDescent<T>* pOsb);
 
            virtual CInternalOptimizerSpecBase<T>* Clone() const override;
 
            virtual const CResult SetWeightDecay(T tWeightDecay = 1e-4f);
            virtual T GetWeightDecay() const;
 
            virtual const CResult EnableSkip1DWeightDecay(bool bSkip1DWeightDecay = true);
            virtual bool IsSkip1DWeightDecayEnabled() const;
 
 
            DeclareGetClassType();
        protected:
            T m_tWeightDecay;
            bool m_bSkip1DWeightDecay;
        };
    }
}
#endif