mxnet
|
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"
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::SignumOptimizer |
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) OptimizerRegistry::__REGISTER__(#Name, [](){return new OptimizerType();}) |
Typedefs | |
typedef std::function< Optimizer *()> | mxnet::cpp::OptimizerCreator |
definition of optimizer
Copyright (c) 2016 by Contributors
#define MXNETCPP_REGISTER_OPTIMIZER | ( | Name, | |
OptimizerType | |||
) | OptimizerRegistry::__REGISTER__(#Name, [](){return new OptimizerType();}) |