Сохранение контрольных точек в формате SavedModel - PullRequest
Новая
       10

Сохранение контрольных точек в формате SavedModel

0 голосов
/ 03 марта 2020

Я использую tenorflow 1.15, у меня есть контрольные точки, сохраненные при обучении модели. Что я хочу сделать, это сохранить модель в формате SavedModel.

Я использую метод восстановления контрольных точек, а затем пытаюсь сохранить модель в формате SavedModel. 

trained_checkpoint_prefix = 'checkpoint_size/checkpoint_size'
export_dir = os.path.join('SavedModel', '0') 
loaded_graph = tf.Graph()
with tf.Session(graph=loaded_graph) as sess:
   # Restore from checkpoint
   loader = tf.train.import_meta_graph(trained_checkpoint_prefix + '.meta')
   loader.restore(sess, trained_checkpoint_prefix)

   # Export checkpoint to SavedModel
   builder = tf.saved_model.builder.SavedModelBuilder(export_dir)
   builder.add_meta_graph_and_variables(sess,[tf.saved_model.tag_constants.TRAINING, tf.saved_model.tag_constants.SERVING],strip_default_attrs=True)
   builder.save()

Это дает мне ValueError: как минимум 2 переменные имеют одинаковые имена

Я просто хочу преобразовать контрольные точки в формат SavedModel.

...