tf.experimental.dtensor.name_based_restore
Восстанавливает DTensor на основе имени из checkpoint_prefix.
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 | Единственная сетка, в которую будут восстановлены все тензоры. |
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/api_docs/python/tf/experimental/dtensor/name_based_restore
