InternalOptimizerStochasticGradientDescent.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "InternalOptimizerGradientDescent.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class CComputationalBase;
 
        template <typename T>
        class CComputationalGraphPlaceholder;
 
        template <typename T>
        class CComputationalGraphAugmentationBase;
 
        template <typename T>
        class CClassEqualizerBase;
 
        template <typename T>
        class CTensor;
 
        template <typename T>
        class FL_EXPORT CInternalOptimizerStochasticGradientDescent : public CInternalOptimizerGradientDescent<T>
        {
        public:
            CInternalOptimizerStochasticGradientDescent();
            CInternalOptimizerStochasticGradientDescent(const CInternalOptimizerStochasticGradientDescent<T>& ob);
 
            virtual ~CInternalOptimizerStochasticGradientDescent();
 
            virtual const CResult Assign(const CInternalOptimizerStochasticGradientDescent<T>& ob);
            virtual const CResult Assign(const CInternalOptimizerStochasticGradientDescent<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 T Validate(int32_t* pI32Iteration = nullptr) override;
 
            virtual const CResult SetMiniBatchSize(int64_t i64Size);
            virtual int64_t GetMiniBatchSize() const;
            virtual int64_t GetTotalIteration() const override;
 
            virtual const CResult InitializeMiniBatch() override;
 
            virtual const CResult EnableAutoSubdivision(bool bAutoSubdivision);
            virtual bool IsAutoSubdivisionEnabled() const;
 
            virtual const CResult EnableIterationMode(bool bIterationMode);
            virtual bool IsIterationModeEnabled() const;
 
            virtual const CResult SetSubdivisionMultiplier(int64_t i64SubdivisionMultiplier = 1);
            virtual int64_t GetSubdivisionMultiplier() const;
 
            DeclareGetClassType();
 
        protected:
            virtual int64_t GetRemainingIteration() const;
            virtual const CResult TerminateCurrentIteration();
 
            virtual int64_t GetMaxMiniBatchSize() const;
            virtual int64_t GetActualMiniBatchSize() const;
 
        protected:
            bool m_bBatchSizeInitialized;
 
            bool m_bAutoSubdivision;
            bool m_bIterationMode;
            int64_t m_i64SubdivisionMultiplier;
        private:
            template<typename T>
            friend class CValidatorBase;
 
            template<typename T>
            friend class CValidatorForClassifier;
 
            template<typename T>
            friend class CValidatorForSemanticSegmentation;
 
            template<typename T>
            friend class CValidatorForObjectDetection;
 
            template<typename T>
            friend class COptimizer;
        };
 
        typedef CInternalOptimizerStochasticGradientDescent<float> CInternalOptimizerStochasticGradientDescentF;
        typedef CInternalOptimizerStochasticGradientDescent<double> CInternalOptimizerStochasticGradientDescentD;
 
        typedef CInternalOptimizerStochasticGradientDescent<float> CInternalOptimizerSGDF;
        typedef CInternalOptimizerStochasticGradientDescent<double> CInternalOptimizerSGDD;
 
        template <typename T>
        using CInternalOptimizerSGD = CInternalOptimizerStochasticGradientDescent<T>;
    }
}
 
#endif