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
