NnetEnsembleTrainerConfig Struct Reference

#include <train-nnet-ensemble.h>

Collaboration diagram for NnetEnsembleTrainerConfig:

Public Member Functions

 NnetEnsembleTrainerConfig ()
void Register (OptionsItf *opts)

Public Attributes

int32 minibatch_size
int32 minibatches_per_phase
double beta

Detailed Description

Definition at line 32 of file train-nnet-ensemble.h.

Constructor & Destructor Documentation

◆ NnetEnsembleTrainerConfig()

Member Function Documentation

◆ Register()

void Register ( OptionsItf opts)

Definition at line 41 of file train-nnet-ensemble.h.

References OptionsItf::Register().

Referenced by main().

41  {
42  opts->Register("minibatch-size", &minibatch_size,
43  "Number of samples per minibatch of training data.");
44  opts->Register("minibatches-per-phase", &minibatches_per_phase,
45  "Number of minibatches to wait before printing training-set "
46  "objective.");
47  opts->Register("beta", &beta,
48  "weight of the second term in the objf, which is the cross-entropy "
49  "between the output posteriors and the averaged posteriors from other nets.");
50  }

Member Data Documentation

◆ beta

double beta

Definition at line 35 of file train-nnet-ensemble.h.

Referenced by NnetEnsembleTrainer::NnetEnsembleTrainer().

◆ minibatch_size

int32 minibatch_size

Definition at line 33 of file train-nnet-ensemble.h.

Referenced by NnetEnsembleTrainer::TrainOnExample().

◆ minibatches_per_phase

int32 minibatches_per_phase

Definition at line 34 of file train-nnet-ensemble.h.

Referenced by NnetEnsembleTrainer::TrainOneMinibatch().

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