ObjectiveFunctionInfo Struct Reference

#include <nnet-training.h>

Collaboration diagram for ObjectiveFunctionInfo:

Public Member Functions
	ObjectiveFunctionInfo ()
void	UpdateStats (const std::string &output_name, int32 minibatches_per_phase, int32 minibatch_counter, BaseFloat this_minibatch_weight, BaseFloat this_minibatch_tot_objf, BaseFloat this_minibatch_tot_aux_objf=0.0)
void	PrintStatsForThisPhase (const std::string &output_name, int32 minibatches_per_phase, int32 phase) const
bool	PrintTotalStats (const std::string &output_name) const

Public Attributes
int32	current_phase
int32	minibatches_this_phase
double	tot_weight
double	tot_objf
double	tot_aux_objf
double	tot_weight_this_phase
double	tot_objf_this_phase
double	tot_aux_objf_this_phase

Detailed Description

Definition at line 123 of file nnet-training.h.

Constructor & Destructor Documentation

ObjectiveFunctionInfo()

ObjectiveFunctionInfo ( )

inline

Definition at line 138 of file nnet-training.h.

                         :
      current_phase(0),
      minibatches_this_phase(0),
      tot_weight(0.0), tot_objf(0.0), tot_aux_objf(0.0),
      tot_weight_this_phase(0.0), tot_objf_this_phase(0.0),
      tot_aux_objf_this_phase(0.0) { }

Member Function Documentation

PrintStatsForThisPhase()

void PrintStatsForThisPhase	(	const std::string &	output_name,
		int32	minibatches_per_phase,
		int32	phase
	)		const

Definition at line 269 of file nnet-training.cc.

References KALDI_LOG.

                       {
  int32 start_minibatch = current_phase * minibatches_per_phase,
      end_minibatch = phase * minibatches_per_phase - 1;

  if (tot_aux_objf_this_phase == 0.0) {
    if (minibatches_per_phase == minibatches_this_phase) {
      KALDI_LOG << "Average objective function for '" << output_name
                << "' for minibatches " << start_minibatch
                << '-' << end_minibatch << " is "
                << (tot_objf_this_phase / tot_weight_this_phase) << " over "
                << tot_weight_this_phase << " frames.";
    } else {
      KALDI_LOG << "Average objective function for '" << output_name
                << " using " << minibatches_this_phase
                << " minibatches in minibatch range " << start_minibatch
                << '-' << end_minibatch << " is "
                << (tot_objf_this_phase / tot_weight_this_phase) << " over "
                << tot_weight_this_phase << " frames.";
    }
  } else {
    BaseFloat objf = (tot_objf_this_phase / tot_weight_this_phase),
        aux_objf = (tot_aux_objf_this_phase / tot_weight_this_phase),
        sum_objf = objf + aux_objf;
    if (minibatches_per_phase == minibatches_this_phase) {
      KALDI_LOG << "Average objective function for '" << output_name
                << "' for minibatches " << start_minibatch
                << '-' << end_minibatch << " is "
                << objf << " + " << aux_objf << " = " << sum_objf
                << " over " << tot_weight_this_phase << " frames.";
    } else {
      KALDI_LOG << "Average objective function for '" << output_name
                << "' using " << minibatches_this_phase
                << " minibatches in  minibatch range " << start_minibatch
                << '-' << end_minibatch << " is "
                << objf << " + " << aux_objf << " = " << sum_objf
                << " over " << tot_weight_this_phase << " frames.";
    }
  }
}

PrintTotalStats()

bool PrintTotalStats

(

const std::string &

output_name

)

const

Definition at line 312 of file nnet-training.cc.

References KALDI_LOG.

Referenced by NnetChainTrainer::PrintTotalStats(), and NnetTrainer::PrintTotalStats().

                                                                       {
  BaseFloat objf = (tot_objf / tot_weight),
        aux_objf = (tot_aux_objf / tot_weight),
        sum_objf = objf + aux_objf;
  if (tot_aux_objf == 0.0) {
    KALDI_LOG << "Overall average objective function for '" << name << "' is "
              << (tot_objf / tot_weight) << " over " << tot_weight << " frames.";
  } else {
    KALDI_LOG << "Overall average objective function for '" << name << "' is "
              << objf << " + " << aux_objf << " = " << sum_objf
              << " over " << tot_weight << " frames.";
  }
  KALDI_LOG << "[this line is to be parsed by a script:] "
            << "log-prob-per-frame="
            << objf;
  return (tot_weight != 0.0);
}

UpdateStats()

void UpdateStats	(	const std::string &	output_name,
		int32	minibatches_per_phase,
		int32	minibatch_counter,
		BaseFloat	this_minibatch_weight,
		BaseFloat	this_minibatch_tot_objf,
		BaseFloat	this_minibatch_tot_aux_objf = 0.0
	)

Definition at line 242 of file nnet-training.cc.

References KALDI_ASSERT.

                                           {
  int32 phase = minibatch_counter / minibatches_per_phase;
  if (phase != current_phase) {
    KALDI_ASSERT(phase > current_phase);
    PrintStatsForThisPhase(output_name, minibatches_per_phase,
                           phase);
    current_phase = phase;
    tot_weight_this_phase = 0.0;
    tot_objf_this_phase = 0.0;
    tot_aux_objf_this_phase = 0.0;
    minibatches_this_phase = 0;
  }
  minibatches_this_phase++;
  tot_weight_this_phase += this_minibatch_weight;
  tot_objf_this_phase += this_minibatch_tot_objf;
  tot_aux_objf_this_phase += this_minibatch_tot_aux_objf;
  tot_weight += this_minibatch_weight;
  tot_objf += this_minibatch_tot_objf;
  tot_aux_objf += this_minibatch_tot_aux_objf;
}

Member Data Documentation

current_phase

int32 current_phase

Definition at line 124 of file nnet-training.h.

minibatches_this_phase

int32 minibatches_this_phase

Definition at line 125 of file nnet-training.h.

tot_aux_objf

double tot_aux_objf

Definition at line 130 of file nnet-training.h.

tot_aux_objf_this_phase

double tot_aux_objf_this_phase

Definition at line 136 of file nnet-training.h.

tot_objf

double tot_objf

Definition at line 129 of file nnet-training.h.

tot_objf_this_phase

double tot_objf_this_phase

Definition at line 135 of file nnet-training.h.

tot_weight

double tot_weight

Definition at line 128 of file nnet-training.h.

tot_weight_this_phase

double tot_weight_this_phase

Definition at line 134 of file nnet-training.h.

---

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

• nnet3/nnet-training.h
• nnet3/nnet-training.cc