Example #1
0
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')
Example #2
0
def setup():
    cm.cublas_init()