mxnet
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Classes | Namespaces | Macros | Typedefs
optimizer.h File Reference

Operator interface of mxnet. More...

#include <dmlc/base.h>
#include <dmlc/logging.h>
#include <dmlc/registry.h>
#include <mshadow/tensor.h>
#include <string>
#include <vector>
#include <utility>
#include "./base.h"
#include "./resource.h"
#include <mxnet/ndarray.h>
Include dependency graph for optimizer.h:

Go to the source code of this file.

Classes

class  mxnet::Optimizer
 
struct  mxnet::OptimizerReg
 Registry entry for Optimizer factory functions. More...
 

Namespaces

 mxnet
 namespace of mxnet
 

Macros

#define MXNET_REGISTER_OPTIMIZER(name, OptimizerType)
 Macro to register Optimizer. More...
 

Typedefs

typedef std::function
< Optimizer *()> 
mxnet::OptimizerFactory
 typedef the factory function of Optimizer More...
 

Detailed Description

Operator interface of mxnet.

Copyright (c) 2015 by Contributors

Author
Junyuan Xie

Macro Definition Documentation

#define MXNET_REGISTER_OPTIMIZER (   name,
  OptimizerType 
)
Value:
DMLC_REGISTRY_REGISTER(::mxnet::OptimizerReg, OptimizerReg, name) \
.set_body([]() { return new OptimizerType(); })
Registry entry for Optimizer factory functions.
Definition: optimizer.h:75

Macro to register Optimizer.

// example of registering a SGD optimizer
MXNET_REGISTER_OPTIMIZER(_SGD, SGDOptimizer)
.describe("Stochastic Gradient Decent optimizer");