mxnet
Classes | Namespaces | Macros | Typedefs
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();})