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