tf.compat.v1.graph_util.remove_training_nodes
Удаляет узлы, не нужные для вывода. (устарело)
tf.compat.v1.graph_util.remove_training_nodes(
input_graph, protected_nodes=None
)
Некоторые узлы, например Identity и CheckNumerics, нужны только во время обучения, поэтому их можно удалять из графов, предназначенных исключительно для вывода. Здесь мы обнаруживаем и удаляем такие узлы, возвращая эквивалентный граф. В частности, узлы CheckNumerics удаляются всегда, а узлы Identity, не участвующие в рёбрах управления, исключаются так, чтобы их входы и выходы соединялись напрямую.
| Аргументы | |
|---|---|
input_graph | Модель для анализа и очистки. |
protected_nodes | Необязательный список имён узлов, которые нужно сохранить в любом случае. Например, это полезно для сохранения выходных узлов Identity. |
| Возвращает | |
|---|---|
| Список узлов, из которого удалены ненужные узлы. |
© 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/remove_training_nodes
