使用Keras和GPU执行超参数搜索的最佳方法?

时间:2018-09-25 17:29:29

标签: scikit-learn keras gpu hyperparameters

我目前想在我拥有的简单Keras神经网络上进行超参数搜索。我当前的计划是使用sklearn GridSearchCV功能,并使用keras.utils.multi_gpu_model将其拆分到我可以访问的2个GPU中。我相信这会很好,但是可能仍需要很长时间才能完成搜索。

有没有更有效的方法?我似乎无法将GridSearch中的n_jobs标志设置为-1,这会使作业并行运行,我只是无法在GPU上执行此操作?

0 个答案:

没有答案