tf.experimental.dtensor.enable_save_as_bf16
Позволяет сохранять и восстанавливать DVariables типа float32 в формате bfloat16.
tf.experimental.dtensor.enable_save_as_bf16(
variables: List[tf.Variable]
)
Метод влияет только на часть DVariable внутри модели и оставляет неизменными переменные/тензоры, не являющиеся DTensor.
| Аргументы | |
|---|---|
variables | Список tf.Variable, которые необходимо включить для сохранения/восстановления в формате bfloat16. Воздействие происходит только на DTensor переменные, поскольку они проходят через d_variables с логикой, специфичной для 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/enable_save_as_bf16
