Оптимизация гиперпараметра для нейронной сети, написанная в керасе - PullRequest
0 голосов
/ 27 апреля 2018

Существует ли библиотека python3, которая оптимизирует гиперпараметры KERAS NN на графическом процессоре?

Я пытался использовать sklearn с оболочкой KerasClassifier, но он использует процессор.

1 Ответ

0 голосов
/ 02 октября 2018

Да, есть. Вы можете использовать Talos , который является решением по оптимизации гиперпараметров специально для моделей Keras. Он поддерживает использование как GPU, так и multi-GPU. Полное раскрытие: я сопровождающий пакета.

...