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