tf.train.load_variable
Возвращает значение тензора заданной переменной в контрольной точке.
tf.train.load_variable(
ckpt_dir_or_file, name
)
Если имя переменной неизвестно, вы можете использовать tf.train.list_variables для проверки всех имён переменных.
Пример использования:
import tensorflow as tf
a = tf.Variable(1.0)
b = tf.Variable(2.0)
ckpt = tf.train.Checkpoint(var_list={'a': a, 'b': b})
ckpt_path = ckpt.save('tmp-ckpt')
var= tf.train.load_variable(
ckpt_path, 'var_list/a/.ATTRIBUTES/VARIABLE_VALUE')
print(var) # 1.0
| Аргументы | |
|---|---|
ckpt_dir_or_file | Каталог с файлами контрольных точек или путь к контрольной точке. |
name | Имя переменной для возврата. |
| Возвращаемое значение | |
|---|---|
Массив NumPy ndarray с копией значения этой переменной. |
© 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/train/load_variable
