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/versions/r2.9/api_docs/python/tf/compat/v1/graph_util/convert_variables_to_constants