tf.experimental.dtensor.name_based_restore
Восстанавливает данные из checkpoint_prefix в DTensors по именам.
tf.experimental.dtensor.name_based_restore(
mesh: tf.experimental.dtensor.Mesh,
checkpoint_prefix: str,
name_tensor_dict: Dict[str, Union[tensor_lib.Tensor, tf_variables.Variable]]
)
Необходимо, чтобы переменные DTensor, из которых восстанавливаются тензоры, уже были инициализированы и имели одинаковую форму/тип данных.
Также в настоящее время поддерживается только восстановление по именам на одном mesh.
| Аргументы | |
|---|---|
mesh | Единственный mesh, к которому будут восстановлены все тензоры. |
checkpoint_prefix | Префикс checkpoint для восстановления. |
name_tensor_dict | Организованный словарь tensor_names к DTensor. Форма/тип данных DTensor должны соответствовать тензорам, сохранённым/восстановленным на данный момент. |
| Возвращаемое значение | |
|---|---|
| Словарь имен и соответствующих им восстановленных значений DTensor. |
© 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/experimental/dtensor/name_based_restore
