from pylab import * import numpy numpy.random.seed(42) # This seed is meaningful. :). import GPULock GPULock.GetGPULock() import cudamat_ext as cm cm.cublas_init() cm.CUDAMatrix.init_random(42) import cudamat as cm2 cm2.CUDAMatrix.init_random(42) import logging from mercurial import ui, hg, localrepo import nnet_train, sys, os import htkdb_cm, shutil import argparse, time from model_averaging import * parser = argparse.ArgumentParser() parser.add_argument('--skip_repo', dest='skip_repo', action='store_true', default=False, help='Do not check repository state') parser.add_argument('--wsj', action='store_true', default=False, help='Are we decoding wsj') group = parser.add_mutually_exclusive_group() group.add_argument('--use_delta', action='store_true', help="Use deltas and accelerations in training") group.add_argument('--no_delta', action='store_true', help="Dont use deltas and accelerations in training") parser.add_argument('--num_frames_per_pt', type=int, default=15, help='Number of frames per point')
def setup(): cm.cublas_init()