tf.compat.v1.graph_util.convert_variables_to_constants
Заменяет все переменные в графе константами с теми же значениями. (устарело)
tf.compat.v1.graph_util.convert_variables_to_constants(
sess, input_graph_def, output_node_names, variable_names_whitelist=None,
variable_names_blacklist=None
)
Если у вас есть обученный граф, содержащий операции Variable, то удобно преобразовать все их в операции Const, хранящие те же значения. Это позволяет полностью описать сеть с помощью одного файла GraphDef и позволяет удалить множество операций, связанных с загрузкой и сохранением переменных.
| Аргументы | |
|---|---|
sess | Активная сессия TensorFlow, содержащая переменные. |
input_graph_def | Объект GraphDef, содержащий сеть. |
output_node_names | Список имен строк для узлов результата графа. |
variable_names_whitelist | Набор имён переменных для преобразования (по умолчанию, все переменные преобразуются). |
variable_names_blacklist | Набор имён переменных, которые не нужно преобразовывать в константы. |
| Возвращаемые значения | |
|---|---|
| GraphDef, содержащий упрощенную версию исходного. |
| Исключения | |
|---|---|
RuntimeError | если найдена операция DT_RESOURCE, родительские переменные которой одновременно находятся в списке запретов И в списке разрешений для заморозки. |
© 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/r2.4/api_docs/python/tf/compat/v1/graph_util/convert_variables_to_constants
