1

Is there a python3 library that optimizes KERAS NN hyperparameters on GPU?

I have tried using sklearn with KerasClassifier wrapper, but it uses cpu.

Mehdi Nellen
  • 8,486
  • 4
  • 33
  • 48
a'-
  • 251
  • 5
  • 14
  • Take a look at [this](https://stackoverflow.com/questions/44802939/hyperparameter-tuning-of-tensorflow-model/44803549#44803549) and [this](https://stackoverflow.com/questions/44181511/hyperparameter-tune-for-tensorflow/44182285#44182285) answers. – jdehesa Apr 27 '18 at 17:12

1 Answers1

0

Yes there is. You can use Talos, which is a hyperparameter optimization solutions specifically for Keras models. It supports both GPU and multi-GPU use. Full disclosure: I'm the maintainer of the package.

mikkokotila
  • 1,403
  • 12
  • 16