并行架构中的PyBrain库

时间:2013-08-02 18:42:23

标签: python parallel-processing neural-network pybrain

是否有人知道pybrain是否在并行架构中工作?如果没有,哪个库可以在多核处理器中与人工神经网络一起使用? 非常感谢!

2 个答案:

答案 0 :(得分:2)

您可以尝试http://deeplearning.net/software/pylearn2/ - 它是使用Theano编写的,因此可以在多核或GPU上运行。

答案 1 :(得分:0)

您询问的内容取决于任务,但 scikit-optimize 能够使用 ask-and-tell 接口并行运行。

P = <parameter space>
optimizer = Optimizer(P)

while True:
    X = optimizer.ask(num_args=8)
    # X now has eight permutations of the parameter space
    Y = wait_for_parallel_evaluate(objective, X)
    # Y now has a list of results for those permutations
    optimizer.tell(X, Y)
    # optimizer uses the results to decide on what parameters it should try next

除了使用这个专用循环,您还可以使用其转储函数序列化优化器,并将其带回调用每个并行结果的 tell 并将其序列化。