#include <optimizer.h>
|
virtual | ~Optimizer () |
| virtual destructor More...
|
|
virtual void | Init (const std::vector< std::pair< std::string, std::string > > &kwargs)=0 |
| Initialize the Optimizer by setting the parameters This function need to be called before all other functions. More...
|
|
virtual void | CreateState (const int index, const NDArray *weight)=0 |
| Create aux state for weigth with index. More...
|
|
virtual void | Update (const int index, NDArray *weight, const NDArray *grad, const float lr, const float wd)=0 |
| Update a weight with gradient. More...
|
|
virtual mxnet::Optimizer::~Optimizer |
( |
| ) |
|
|
inlinevirtual |
static Optimizer* mxnet::Optimizer::Create |
( |
const char * |
type_name | ) |
|
|
static |
virtual void mxnet::Optimizer::CreateState |
( |
const int |
index, |
|
|
const NDArray * |
weight |
|
) |
| |
|
pure virtual |
Create aux state for weigth with index.
- Parameters
-
index | the unique index for the weight. |
weight | the NDArray to associate created state to. |
virtual void mxnet::Optimizer::Init |
( |
const std::vector< std::pair< std::string, std::string > > & |
kwargs | ) |
|
|
pure virtual |
Initialize the Optimizer by setting the parameters This function need to be called before all other functions.
- Parameters
-
kwargs | the keyword arguments parameters |
virtual void mxnet::Optimizer::Update |
( |
const int |
index, |
|
|
NDArray * |
weight, |
|
|
const NDArray * |
grad, |
|
|
const float |
lr, |
|
|
const float |
wd |
|
) |
| |
|
pure virtual |
Update a weight with gradient.
- Parameters
-
index | the unique index for the weight. |
weight | the weight to update. |
grad | gradient for the weight. |
lr | learning rate for this update. |
wd | weight decay for this update. |
The documentation for this class was generated from the following file: