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()

NnetShrinkConfig ( )

inline

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

: num_bfgs_iters(10), initial_step(0.1) { }

Member Function Documentation

Register()

void Register ( OptionsItf *  opts )

inline

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

References OptionsItf::Register(), and kaldi::nnet2::ShrinkNnet().

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

Member Data Documentation

initial_step

BaseFloat initial_step

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

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

num_bfgs_iters

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:

• nnet2/shrink-nnet.h