Распределение памяти графического процессора Tensorflow влияет на точность модели - PullRequest
0 голосов
/ 10 октября 2018

У меня есть нейронная сеть в Keras 2.2.2, Tensorflow 1.10.0, Ubuntu 18, GTX1080Ti.Я обучил модель, используя а) всю память графического процессора и б) примерно половину (0,47) памяти.Я получаю резко разные результаты между этими случаями.Полная память - хороший результат.Половина памяти - плохо.Я повторил эти тесты, и результаты соответствуют.Код модели слишком сложен, чтобы показывать здесь, но мне интересно, является ли это известным общим эффектом.Я установил распределение памяти с помощью этого кода:

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.47
session = tf.Session(config=config)
K.set_session(session)

Размер пакета вычисляется в коде во время выполнения и не зависит от выделения памяти графическим процессором.

...