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, у которой переменные-предки одновременно находятся в списке запрещенных и в списке разрешенных для фиксации. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/graph_util/convert_variables_to_constants