This commit is contained in:
Alexander Pivovarov 2019-04-16 13:28:37 -07:00 коммит произвёл Tianqi Chen
Родитель 7ee5cbac6d
Коммит 2b24e23fbf
1 изменённых файлов: 3 добавлений и 5 удалений

Просмотреть файл

@ -85,11 +85,9 @@ class TFParser(object):
tags,
self._model_dir)
output_names = set()
for k in meta_graph_def.signature_def.keys():
outputs_tensor_info = meta_graph_def.signature_def[k].outputs
for output_tensor in outputs_tensor_info.values():
output_names.add(output_tensor.name)
output_names = [i.replace(":0", "") for i in output_names]
for sig_def in meta_graph_def.signature_def.values():
for output_tensor in sig_def.outputs.values():
output_names.add(output_tensor.name.replace(":0", ""))
return ",".join(output_names)
def _load_saved_model(self):