예제 #1
0
파일: lm.py 프로젝트: philip30/chainn
def init_program_state(args):
    global xp, USE_LSTM, USE_GPU, HIDDEN_SIZE
    xp          = UF.select_wrapper(not args.use_cpu)
    HIDDEN_SIZE = args.hidden_size
    USE_GPU     = not args.use_cpu
    USE_LSTM    = args.model == "lstm"
예제 #2
0
파일: rnn-pos.py 프로젝트: philip30/chainn
def init_wrapper(use_gpu):
    global xp
    xp = UF.select_wrapper(use_gpu)