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

Member Function Documentation

◆ Register()

void Register ( OptionsItf po)

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

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

Referenced by main().

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

Member Data Documentation

◆ edge_minibatch_size

int32 edge_minibatch_size

◆ ensure_exact_final_context

bool ensure_exact_final_context

◆ minibatch_size

int32 minibatch_size

◆ partial_minibatch_factor

BaseFloat partial_minibatch_factor

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