tf.contrib.checkpoint.split_dependency
Создаёт несколько зависимостей с синхронизированной сохранением/восстановлением.
tf.contrib.checkpoint.split_dependency(
component_names, component_dtypes, fill_save_buffer_fn,
consume_restore_buffer_fn, device
)
Полезно, когда один оператор производит Tensors, которые должны быть сохранены под разными объектами, или когда Tensors, сохранённые с множеством разных объектов, должны быть восстановлены вместе как входные данные для одного оператора (т. е. объект, использующий один объединённый оператор, может быть заменён подграфом объектов, и эти две программы совместимы с контрольными точками).
| Аргументы | |
|---|---|
component_names | Последовательность имён для разделенных зависимостей. fill_save_buffer_fn должен добавить эти ключи в словарь, который ему передаётся, а consume_restore_buffer_fn получит словарь с этими ключами. |
component_dtypes | Типы данных для Tensors, которые сохраняются и восстанавливаются, последовательность, соответствующая component_names. |
fill_save_buffer_fn | Функция, которая принимает пустой словарь в качестве аргумента и добавляет Tensors с component_names в качестве ключей. Эти Tensors будут сохранены так, как будто они являются отдельными переменными. |
consume_restore_buffer_fn | Функция, которая принимает словарь с component_names в качестве ключей, сопоставленных с восстановленными отдельными Tensors, и возвращает оператор восстановления (или, если выполняется увлечение, выполняет восстановление и может вернуть None). |
device | Устройство, на котором необходимо выполнять операции сохранения и восстановления. |
| Возвращаемые значения | |
|---|---|
| Словарь, сопоставляющий имена с объектами Trackable. Если один достижим из объекта в качестве зависимости, то и другие тоже; добавление зависимостей к некоторым, но не ко всем объектам, приведёт к ошибкам. |
© 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/r1.15/api_docs/python/tf/contrib/checkpoint/split_dependency
