Keras fit_generator не использует все ядра на COLAB GPU - PullRequest
0 голосов
/ 17 июня 2019

Я использую модель Keras с бэкэндом тензорного потока.Он работает на настольном компьютере - хотя все 16 ядер моей машины с Windows не используются.

Я перенес его в COLAB Googles и запустил там.Я использую fit_genrator от keras, поскольку в наборе данных содержится 34 млн. Тренировочных пар.Проблема в том, что ускорение от ядер 2000+ полностью отсутствует - в основном улучшений нет.Кто-нибудь сталкивался с этим?

Я видел здесь обсуждение подобной проблемы https://github.com/keras-team/keras/issues/7732

Сам код является реализацией скипграмм -

SkipGram.fit_generator (batch_manager,эпохи = эпохи, многословные = 1, use_multiprocessing = True, max_queue_size = 50)

Спасибо.

...