tf.contrib.eager.restore_variables_on_create

Менеджер контекста, восстанавливающий переменные при создании.

@contextlib.contextmanager
tf.contrib.eager.restore_variables_on_create(
    save_path, map_func=None
)

Если save_path равно None (например, нет контрольной точки), ничего не происходит. В противном случае все значения из контрольной точки предварительно загружаются. Когда соответствующая переменная создаётся впервые, значение из контрольной точки присваивается переменной.

with restore_variables_on_create(
    tf.train.latest_checkpoint(checkpoint_dir)):

Аргументы

save_path Префикс файла контрольной точки.

map_func Функция, которая, получив имя переменной в качестве аргумента, возвращает имя переменной в контрольной точке для восстановления. Если None, используется переменная с тем же именем в контрольной точке для восстановления. Ошибка, если сопоставленное имя переменной отсутствует в контрольной точке.

Аргументы
`save_path`	Префикс файла контрольной точки.
`map_func`	Функция, которая, получив имя переменной в качестве аргумента, возвращает имя переменной в контрольной точке для восстановления. Если None, используется переменная с тем же именем в контрольной точке для восстановления. Ошибка, если сопоставленное имя переменной отсутствует в контрольной точке.

Возвращает:

Ничего.

Исключения
`NotFoundError`	Если переменная не найдена в контрольной точке.
`ValueError`	Если не используется в режиме eager или map_func не является вызываемой функцией.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/eager/restore_variables_on_create