mlpack
|
Public Member Functions | |
GradientZeroVisitor () | |
Set the gradient to zero for the given module. More... | |
template<typename LayerType > | |
void | operator() (LayerType *layer) const |
Set the gradient to zero. | |
void | operator() (MoreTypes layer) const |
|
inline |
Set the gradient to zero for the given module.
GradientZeroVisitor visitor class.