tf.train.export_meta_graph
Возвращает MetaGraphDef протокол.
tf.train.export_meta_graph(
filename=None, meta_info_def=None, graph_def=None, saver_def=None,
collection_list=None, as_text=False, graph=None, export_scope=None,
clear_devices=False, clear_extraneous_savers=False, strip_default_attrs=False,
save_debug_info=False, **kwargs
)
При необходимости записывает его в файл filename.
Эта функция экспортирует граф, объект saver и объекты коллекций в MetaGraphDef протокол с целью его импорта в другое время или место для возобновления обучения, выполнения вывода или создания подграфа.
| Аргументы | |
|---|---|
filename | Необязательное имя файла, включая путь для записи сгенерированного MetaGraphDef протокола. |
meta_info_def | MetaInfoDef протокол. |
graph_def | GraphDef протокол. |
saver_def | SaverDef протокол. |
collection_list | Список строковых ключей для сбора. |
as_text | Если True, записывает MetaGraphDef в виде ASCII-протокола. |
graph | Объект Graph для экспорта. Если None, используется граф по умолчанию. |
export_scope | Необязательный string. Имя области именования, под которой извлекается подграф. Имя области именования будет удалено из определений узлов для простоты импорта позже в новые области именования. Если None, экспортируется весь граф. graph_def и export_scope нельзя оба указать. |
clear_devices | Нужно ли очистить поле устройства для Operation или Tensor при экспорте. |
clear_extraneous_savers | Удалить любые данные, относящиеся к Saver (как операции Save/Restore, так и SaverDef), которые не связаны с предоставленным SaverDef, из графа. |
strip_default_attrs | Булево. Если True, атрибуты с значениями по умолчанию будут удалены из NodeDef. Для подробного руководства см. Удаление атрибутов со значениями по умолчанию. |
save_debug_info | Если True, сохранить GraphDebugInfo в отдельный файл, который находится в той же директории, что и filename, и с _debug добавлен перед расширением файла. |
**kwargs | Необязательные аргументы с ключами. |
| Возвращаемое значение | |
|---|---|
A MetaGraphDef протокол. |
| Исключения | |
|---|---|
ValueError | Когда размер GraphDef превышает 2 ГБ. |
RuntimeError | Если вызов происходит при включенном режиме eager execution. |
Совместимость с режимом Eager
Экспорт/импорт метаграфов не поддерживается, если оба graph_def и graph не предоставлены. Граф не существует, когда режим eager execution включен.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/train/export_meta_graph