function_pkl_test.py 文件源码

def test_one_process(gpu=0):
    theano.gpuarray.use("cuda" + str(gpu))

    f_train, train_name = build_train_func()
    pickle_func(f_train)
    f_unpkl, unpkl_name = unpickle_func()

    test_the_function(f_train, train_name)
    test_the_function(f_unpkl, unpkl_name)