InternalOptimizerSpecStochasticGradientDescent.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "InternalOptimizerSpecGradientDescent.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CInternalOptimizerSpecStochasticGradientDescent : public CInternalOptimizerSpecGradientDescent<T>
        {
        public:
            CInternalOptimizerSpecStochasticGradientDescent();
            CInternalOptimizerSpecStochasticGradientDescent(const CInternalOptimizerSpecStochasticGradientDescent<T>& osb);
            CInternalOptimizerSpecStochasticGradientDescent(const CInternalOptimizerSpecStochasticGradientDescent<T>* pOsb);
 
            virtual ~CInternalOptimizerSpecStochasticGradientDescent();
 
            virtual const CResult Assign(const CInternalOptimizerSpecStochasticGradientDescent<T>& osb);
            virtual const CResult Assign(const CInternalOptimizerSpecStochasticGradientDescent<T>* pOsb);
 
            virtual CInternalOptimizerSpecBase<T>* Clone() const override;
 
            virtual const CResult EnableAutoSubdivision(bool bAutoSubdivision = true);
            virtual bool IsAutoSubdivisionEnabled() const;
 
            virtual const CResult SetMiniBatchSize(int64_t i64MiniBatchSize);
            virtual int64_t GetMiniBatchSize() const;
 
            virtual const CResult SetMiniBatchIteration(int64_t i64Size);
            virtual int64_t GetMiniBatchIteration() const;
 
            virtual const CResult EnableIterationMode(bool bIterationMode);
            virtual bool IsIterationModeEnabled() const;
 
            DeclareGetClassType();
        protected:
            bool m_bAutoSubdivision;
            int64_t m_i64MiniBatchSize;
            int64_t m_i64IterationSize;
            bool m_bIterationMode;
        };
    }
}
#endif