Различные модели TensorFlow / Keras на разных графических процессорах, повторное обучение - PullRequest
0 голосов
/ 02 марта 2019

У меня есть две модели, и я хочу несколько раз обучить их следующим образом:

step1.model1.fit (data1) -> использовать графический процессор # 1, создать output1

step2.model2.fit (data2) -> использовать графический процессор # 2, создать output2

step3.newdata1, newdata2 = функция (выход1, выход2)

step4.model1.fit (newdata1) -> использовать графический процессор # 1

step5.model2.fit (newdata2) -> использовать GPU # 2 .....

Причина обучения отдельной модели по отдельности связана с ограничением памяти.Я могу реализовать это к этому post , если я хочу переопределить модель в новом сеансе в шаге 4/5, и загрузить веса, сохраненные в шаге 1/2.Тем не менее, модель и вес большие.Итак, я хотел бы определить модель один раз на шаге 1/2 и просто продолжить обучение на шаге 4/5.

Интересно, есть ли способ реализовать это?Любое предложение будет оценено.

...