标签: scikit-learn keras gpu hyperparameters
我目前想在我拥有的简单Keras神经网络上进行超参数搜索。我当前的计划是使用sklearn GridSearchCV功能,并使用keras.utils.multi_gpu_model将其拆分到我可以访问的2个GPU中。我相信这会很好,但是可能仍需要很长时间才能完成搜索。
有没有更有效的方法?我似乎无法将GridSearch中的n_jobs标志设置为-1,这会使作业并行运行,我只是无法在GPU上执行此操作?