InternalOptimizerMuonGradientDescent.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "InternalOptimizerAdamWGradientDescent.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CInternalOptimizerMuonGradientDescent : public CInternalOptimizerAdamWGradientDescent<T>
        {
        public:
            CInternalOptimizerMuonGradientDescent();
            CInternalOptimizerMuonGradientDescent(const CInternalOptimizerMuonGradientDescent<T>& ob);
 
            virtual ~CInternalOptimizerMuonGradientDescent();
 
            virtual const CResult Assign(const CInternalOptimizerMuonGradientDescent<T>& ob);
            virtual const CResult Assign(const CInternalOptimizerMuonGradientDescent<T>* pOb);
 
            virtual CInternalOptimizerBase<T>* Clone() const override;
 
            virtual const CResult UpdateWeights() override;
 
            virtual int64_t GetRequiredDedicatedMemory(int64_t i64BatchSize) const override;
 
 
            DeclareGetClassType();
 
        protected:
 
            virtual void CreateCumulativeGradients();
        };
 
 
        typedef CInternalOptimizerMuonGradientDescent<float> CInternalOptimizerMuonGradientDescentF;
        typedef CInternalOptimizerMuonGradientDescent<double> CInternalOptimizerMuonGradientDescentD;
 
        typedef CInternalOptimizerMuonGradientDescent<float> CInternalOptimizerMuonGDF;
        typedef CInternalOptimizerMuonGradientDescent<double> CInternalOptimizerMuonGDD;
 
        template <typename T>
        using CInternalOptimizerMuonGD = CInternalOptimizerMuonGradientDescent<T>;
    }
}
 
#endif