用keras编写的神经网络超参数优化

时间:2018-04-27 17:07:52

标签: tensorflow optimization keras hyperparameters talos

是否有一个python3库可以优化GPU上的KERAS NN超参数?

我尝试过使用带有KerasClassifier包装器的sklearn,但是它使用了cpu。

1 个答案:

答案 0 :(得分:0)

是的。您可以使用Talos,这是专门用于Keras模型的超参数优化解决方案。它支持GPU和多GPU使用。全面披露:我是软件包的维护者。