NnetBatchComputerOptions Struct Reference

#include <nnet-batch-compute.h>

Inheritance diagram for NnetBatchComputerOptions:

Collaboration diagram for NnetBatchComputerOptions:

Public Member Functions
	NnetBatchComputerOptions ()
void	Register (OptionsItf *po)
Public Member Functions inherited from NnetSimpleComputationOptions
	NnetSimpleComputationOptions ()
void	Register (OptionsItf *opts)
void	CheckAndFixConfigs (int32 nnet_modulus)

Public Attributes
int32	minibatch_size
int32	edge_minibatch_size
bool	ensure_exact_final_context
BaseFloat	partial_minibatch_factor
Public Attributes inherited from NnetSimpleComputationOptions
int32	extra_left_context
int32	extra_right_context
int32	extra_left_context_initial
int32	extra_right_context_final
int32	frame_subsampling_factor
int32	frames_per_chunk
BaseFloat	acoustic_scale
bool	debug_computation
NnetOptimizeOptions	optimize_config
NnetComputeOptions	compute_config
CachingOptimizingCompilerOptions	compiler_config

Detailed Description

Definition at line 144 of file nnet-batch-compute.h.

Constructor & Destructor Documentation

NnetBatchComputerOptions()

NnetBatchComputerOptions ( )

inline

Definition at line 150 of file nnet-batch-compute.h.

                            : minibatch_size(128),
                              edge_minibatch_size(32),
                              ensure_exact_final_context(false),
                              partial_minibatch_factor(0.5) {
  }

Member Function Documentation

Register()

void Register ( OptionsItf *  po )

inline

Definition at line 156 of file nnet-batch-compute.h.

References kaldi::nnet3::MergeTaskOutput(), NnetInferenceTask::output, OptionsItf::Register(), and NnetSimpleComputationOptions::Register().

Referenced by main().

                                {
    NnetSimpleComputationOptions::Register(po);
    po->Register("minibatch-size", &minibatch_size, "Number of chunks per "
                 "minibatch (see also edge-minibatch-size)");
    po->Register("edge-minibatch-size", &edge_minibatch_size, "Number of "
                 "chunks per minibatch: this applies to chunks at the "
                 "beginnings and ends of utterances, in cases (such as "
                 "recurrent models) when the computation would be different "
                 "from the usual one.");
    po->Register("ensure-exact-final-context", &ensure_exact_final_context,
                 "If true, for utterances shorter than --frames-per-chunk, "
                 "use exact-length, special computations.  If false, "
                 "pad with repeats of the last frame.  Would only affect "
                 "the output for backwards-recurrent models, but would "
                 "negatively impact speed in all cases.");
    po->Register("partial-minibatch-factor", &partial_minibatch_factor,
                 "Factor that controls how small partial minibatches will be "
                 "they become necessary.  We will potentially do the computation "
                 "for sizes: int(partial_minibatch_factor^n * minibatch_size "
                 ", for n = 0, 1, 2....  Set it to 0.0 if you want to use "
                 "only the specified minibatch sizes.");
  }

Member Data Documentation

edge_minibatch_size

int32 edge_minibatch_size

Definition at line 146 of file nnet-batch-compute.h.

Referenced by NnetBatchComputer::GetMinibatchSize(), and NnetBatchComputer::NnetBatchComputer().

ensure_exact_final_context

bool ensure_exact_final_context

Definition at line 147 of file nnet-batch-compute.h.

Referenced by kaldi::nnet3::utterance_splitting::GetOutputFrameInfoForTasks().

minibatch_size

int32 minibatch_size

Definition at line 145 of file nnet-batch-compute.h.

Referenced by NnetBatchComputer::GetMinibatchSize(), and NnetBatchComputer::NnetBatchComputer().

partial_minibatch_factor

BaseFloat partial_minibatch_factor

Definition at line 148 of file nnet-batch-compute.h.

Referenced by NnetBatchComputer::GetActualMinibatchSize(), and NnetBatchComputer::NnetBatchComputer().

---

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

• nnet3/nnet-batch-compute.h