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[ops.Tensor, tf_variables.Variable]]
)
Необходимо, чтобы переменные DTensor, которые будут восстановлены, уже были инициализированы и имели одинаковую форму/тип данных.
Кроме того, в настоящее время мы поддерживаем только восстановление по именам на одной сетке.
| Аргументы | |
|---|---|
mesh | Единственная сетка, на которую будут восстановлены все тензоры. |
checkpoint_prefix | Префикс чекпойнта, который нужно восстановить. |
name_tensor_dict | Упорядоченный словарь имен тензоров к 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/versions/r2.9/api_docs/python/tf/experimental/dtensor/name_based_restore
