function_pkl_test.py 文件源码

def sequence_worker(rank, n_gpu, barrier, function_maker):
    theano.gpuarray.use("cuda" + str(rank))
    # maybe master makes the function
    barrier.wait()
    f_train, name = function_maker(rank=rank)  # maybe unpickle
    barrier.wait()
    for i in range(n_gpu):
        time.sleep(1)
        barrier.wait()
        if i == rank:
            test_the_function(f_train, name=name, rank=rank)