nnet-optimize-test.cc File Reference

#include "nnet3/nnet-nnet.h"
#include "nnet3/nnet-compile.h"
#include "nnet3/nnet-analyze.h"
#include "nnet3/nnet-test-utils.h"
#include "nnet3/nnet-optimize.h"
#include "nnet3/nnet-compute.h"

Include dependency graph for nnet-optimize-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

Macros
#define	KALDI_SUCCFAIL(b)   ((b) ? "SUCCESS" : "FAILURE")

Functions
static bool	UnitTestNnetOptimizeWithOptions (int32 srand_seed, NnetOptimizeOptions opt_config, CachingOptimizingCompilerOptions compiler_config)
static void	UnitTestNnetOptimizeInternal (int32 srand_seed)
static void	UnitTestNnetOptimize ()
int	main ()

Macro Definition Documentation

KALDI_SUCCFAIL

#define KALDI_SUCCFAIL

(

b

)

((b) ? "SUCCESS" : "FAILURE")

Referenced by kaldi::nnet3::UnitTestNnetOptimizeInternal().

Function Documentation

main()

int main

(

)

Definition at line 308 of file nnet-optimize-test.cc.

References KALDI_LOG, kaldi::SetVerboseLevel(), and kaldi::nnet3::UnitTestNnetOptimize().

           {
  using namespace kaldi;
  using namespace kaldi::nnet3;
  SetVerboseLevel(3);

#if HAVE_CUDA == 1
  CuDevice::Instantiate().SetDebugStrideMode(true);
  CuDevice::Instantiate().SelectGpuId("no");
  UnitTestNnetOptimize();
  CuDevice::Instantiate().SelectGpuId("yes");
#endif
  UnitTestNnetOptimize();

  KALDI_LOG << "Nnet tests succeeded.";

  return 0;
}