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. |
| Возвращаемые значения | |
|---|---|
| Список узлов с удаленными ненужными. |
© 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/remove_training_nodes
