Is there a python3 library that optimizes KERAS NN hyperparameters on GPU?
I have tried using sklearn with KerasClassifier wrapper, but it uses cpu.
Is there a python3 library that optimizes KERAS NN hyperparameters on GPU?
I have tried using sklearn with KerasClassifier wrapper, but it uses cpu.
Copyright © 2021 Jogjafile Inc.
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.