Загрузите замороженную модель TensorFlow (как файл * .pb) и продолжайте тренировать ее - PullRequest
0 голосов
/ 16 апреля 2019

Я работаю над репозиторием Uber Atari-model-zoo (https://github.com/uber-research/atari-model-zoo).

) В основном я пытаюсь загрузить замороженную модель с ранней стадии обучения и продолжаю тренировать еелокально, при выполнении некоторых действий над графиком (прямая пересылка время от времени и т. д.).

Uber предоставляет замороженные модели в виде файлов * .pb с разных этапов обучения, но я не понимаю, как это сделатьпересоздать график и загрузить в него веса и другие параметры метаданных.

В идеале я хочу сделать что-то вроде:

  1. m = import_model ('.pb ')

  2. для эпохи:

    2,1 для партии:

    2.1.1 sess.run(m.training_op, feed_dict={x,y})
    

спасибо.

...