GPU в цикле for - PullRequest
Новая
       32

GPU в цикле for

1 голос
/ 27 мая 2020

Я пытаюсь выполнить для l oop функцию, которая включает нейронную сеть. nn_f - это функция, которая обучает нейронную сеть с бэкэндом Keras, установленным как GPU: 

import os 
os.environ["KERAS_BACKEND"] = "plaidml.keras.backend" # was #Tensorflow
os.environ["PLAIDML_EXPERIMENTAL"] = "1"  # i.e. y/n question in terminal : "1" represent yes 
os.environ["PLAIDML_DEVICE_IDS"] = "opencl_amd_gfx1010.0"

Запуск индивидуума nn_f достаточно быстр.

Однако я считаю, что l oop работает намного медленнее, чем ожидалось, и мне интересно, как ускорить весь процесс. 

func = partial(nn_f,batch_size, epochs ...)
R_2_l = [ ]
for x, y in zip(param_l[0], param_l[1]):

           R_2_l.append(func(x,y))
...