function_pkl_test.py 文件源码

def simultaneous_worker(rank, function_maker, barrier, bar_loop):
    theano.gpuarray.use("cuda" + str(rank))
    # maybe master makes the function
    barrier.wait()
    f_train, name = function_maker(rank)
    barrier.wait()
    test_the_function(f_train, name=name, rank=rank, barrier=barrier, bar_loop=bar_loop)