nnet-component-test.cc File Reference

#include "nnet3/nnet-nnet.h"
#include "nnet3/nnet-simple-component.h"
#include "nnet3/nnet-test-utils.h"

Include dependency graph for nnet-component-test.cc:

Go to the source code of this file.

Namespaces
	kaldi
	This code computes Goodness of Pronunciation (GOP) and extracts phone-level pronunciation feature for mispronunciations detection tasks, the reference:
	kaldi::nnet3

Functions
static void	ResetSeed (int32 rand_seed, const Component &c)
bool	CheckStringsApproxEqual (const std::string &a, const std::string &b, int32 tolerance=3)
void	TestNnetComponentIo (Component *c)
void	TestNnetComponentCopy (Component *c)
void	TestNnetComponentAddScale (Component *c)
void	TestNnetComponentVectorizeUnVectorize (Component *c)
void	TestNnetComponentUpdatable (Component *c)
ComponentPrecomputedIndexes *	GetPrecomputedIndexes (const Component &c, int32 num_rows)
void	TestSimpleComponentPropagateProperties (const Component &c)
bool	TestSimpleComponentDataDerivative (const Component &c, BaseFloat perturb_delta)
bool	TestSimpleComponentModelDerivative (const Component &c, BaseFloat perturb_delta, bool test_derivative)
void	UnitTestNnetComponent ()
int	main ()

Function Documentation

main()

int main

(

)

Definition at line 524 of file nnet-component-test.cc.

References KALDI_LOG, and kaldi::nnet3::UnitTestNnetComponent().

           {
  using namespace kaldi;
  using namespace kaldi::nnet3;
#if HAVE_CUDA == 1
  kaldi::int32 loop = 0;
  for (loop = 0; loop < 2; loop++) {
    //CuDevice::Instantiate().SetDebugStrideMode(true);
    if (loop == 0)
      CuDevice::Instantiate().SelectGpuId("no");
    else
      CuDevice::Instantiate().SelectGpuId("yes");
#endif
    UnitTestNnetComponent();
#if HAVE_CUDA == 1
  } // No for loop if 'HAVE_CUDA != 1',
  CuDevice::Instantiate().PrintProfile();
#endif
  KALDI_LOG << "Nnet component tests succeeded.";

  return 0;
}