All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
NnetShrinkConfig Struct Reference

Configuration class that controls neural net "shrinkage" which is actually a scaling on the parameters of each of the updatable layers. More...

#include <shrink-nnet.h>

Collaboration diagram for NnetShrinkConfig:

Public Member Functions

 NnetShrinkConfig ()
 
void Register (OptionsItf *opts)
 

Public Attributes

int32 num_bfgs_iters
 
BaseFloat initial_step
 

Detailed Description

Configuration class that controls neural net "shrinkage" which is actually a scaling on the parameters of each of the updatable layers.

Definition at line 33 of file shrink-nnet.h.

Constructor & Destructor Documentation

NnetShrinkConfig ( )
inline

Definition at line 41 of file shrink-nnet.h.

Member Function Documentation

void Register ( OptionsItf opts)
inline

Definition at line 42 of file shrink-nnet.h.

References OptionsItf::Register().

42  {
43  opts->Register("num-bfgs-iters", &num_bfgs_iters, "Number of iterations of "
44  "BFGS to use when optimizing shrinkage parameters");
45  opts->Register("initial-step", &initial_step, "Parameter in the optimization, "
46  "used to set the initial step length");
47  }

Member Data Documentation

BaseFloat initial_step

Definition at line 39 of file shrink-nnet.h.

Referenced by kaldi::nnet2::ShrinkNnet().

int32 num_bfgs_iters

Definition at line 34 of file shrink-nnet.h.

Referenced by kaldi::nnet2::ShrinkNnet().


The documentation for this struct was generated from the following file: