Optimizer interface.
More...
#include <optimizer.h>
◆ Optimizer()
mxnet::cpp::Optimizer::Optimizer |
( |
unsigned |
begin_num_update | ) |
|
|
explicit |
constructor
- Parameters
-
beign_num_update | The initial number of updates |
◆ ~Optimizer()
virtual mxnet::cpp::Optimizer::~Optimizer |
( |
| ) |
|
|
virtual |
◆ CreateState_()
virtual void mxnet::cpp::Optimizer::CreateState_ |
( |
int |
index, |
|
|
NDArray |
weight |
|
) |
| |
|
protectedvirtual |
◆ GetLR_()
float mxnet::cpp::Optimizer::GetLR_ |
( |
int |
index | ) |
|
|
protected |
◆ GetParamKeys_()
const std::vector<const char*> mxnet::cpp::Optimizer::GetParamKeys_ |
( |
| ) |
const |
|
protected |
◆ GetParamValues_()
const std::vector<const char*> mxnet::cpp::Optimizer::GetParamValues_ |
( |
| ) |
const |
|
protected |
◆ GetType()
virtual std::string mxnet::cpp::Optimizer::GetType |
( |
| ) |
const |
|
pure virtual |
◆ GetWD_()
float mxnet::cpp::Optimizer::GetWD_ |
( |
int |
index | ) |
|
|
protected |
◆ op_map()
static OpMap*& mxnet::cpp::Optimizer::op_map |
( |
| ) |
|
|
staticprotected |
◆ Serialize()
std::string mxnet::cpp::Optimizer::Serialize |
( |
| ) |
const |
Serialize the optimizer parameters to a string.
- Returns
- serialization
◆ SetLRScheduler()
set the lr scheduler
- Parameters
-
lrScheduler | lr scheduler used for this optimizer |
- Returns
- reference if self
◆ SetParam()
template<typename T >
Optimizer* mxnet::cpp::Optimizer::SetParam |
( |
const std::string & |
name, |
|
|
const T & |
value |
|
) |
| |
|
inline |
set config parameters
- Parameters
-
name | name of the config parameter |
value | value of the config parameter |
- Returns
- reference of self
◆ Update()
virtual void mxnet::cpp::Optimizer::Update |
( |
int |
index, |
|
|
NDArray |
weight, |
|
|
NDArray |
grad |
|
) |
| |
|
pure virtual |
◆ UpdateCount_()
unsigned mxnet::cpp::Optimizer::UpdateCount_ |
( |
int |
index | ) |
|
|
protected |
◆ begin_num_update_
unsigned mxnet::cpp::Optimizer::begin_num_update_ |
|
protected |
◆ count_
std::map<int, unsigned> mxnet::cpp::Optimizer::count_ |
|
protected |
◆ lrScheduler_
std::unique_ptr<LRScheduler> mxnet::cpp::Optimizer::lrScheduler_ = nullptr |
|
protected |
◆ num_update_
unsigned mxnet::cpp::Optimizer::num_update_ |
|
protected |
◆ params_
std::map<std::string, std::string> mxnet::cpp::Optimizer::params_ |
|
protected |
The documentation for this class was generated from the following file: