optimizer.h File Reference

definition of optimizer More...

#include <map>
#include <vector>
#include <string>
#include <memory>
#include <functional>
#include "mxnet-cpp/base.h"
#include "dmlc/logging.h"
#include "mxnet-cpp/ndarray.h"
#include "mxnet-cpp/op_map.h"
#include "mxnet-cpp/lr_scheduler.h"

Include dependency graph for optimizer.h:

Go to the source code of this file.

Classes
class	mxnet::cpp::Optimizer
	Optimizer interface. More...
class	mxnet::cpp::OptimizerRegistry
class	mxnet::cpp::SGDOptimizer
class	mxnet::cpp::RMSPropOptimizer
class	mxnet::cpp::AdamOptimizer
class	mxnet::cpp::AdaGradOptimizer
class	mxnet::cpp::AdaDeltaOptimizer

Namespaces
	mxnet
	namespace of mxnet
	mxnet::cpp

Macros
#define	MXNETCPP_REGISTER_OPTIMIZER(Name, OptimizerType)

Typedefs
typedef std::function< Optimizer *()>	mxnet::cpp::OptimizerCreator

Detailed Description

definition of optimizer

Author

Chuntao Hong, Zhang Chen

Macro Definition Documentation

#define MXNETCPP_REGISTER_OPTIMIZER	(		Name,
			OptimizerType
	)

Value:

static int __make_ ## OptimizerType ## _ ## Name ## __ = \
       OptimizerRegistry::__REGISTER__(#Name, [](){return new OptimizerType();})