tf.compat.v1.train.Saver

Сохраняет и восстанавливает переменные.

tf.compat.v1.train.Saver(
    var_list=None, reshape=False, sharded=False, max_to_keep=5,
    keep_checkpoint_every_n_hours=10000.0, name=None, restore_sequentially=False,
    saver_def=None, builder=None, defer_build=False, allow_empty=False,
    write_version=tf.train.SaverDef.V2, pad_step_number=False,
    save_relative_paths=False, filename=None
)

См. Переменные для обзора переменных, сохранения и восстановления.

Класс Saver добавляет операции для сохранения и восстановления переменных в и из чекпоинтов. Он также предоставляет удобные методы для выполнения этих операций.

Чекпоинты — это двоичные файлы в собственном формате, которые сопоставляют имена переменных со значениями тензоров. Лучший способ проверить содержимое чекпоинта — загрузить его с помощью Saver.

Сохранители могут автоматически нумеровать имена файлов чекпоинтов с помощью предоставленного счетчика. Это позволяет хранить несколько чекпоинтов на разных шагах во время обучения модели. Например, можно нумеровать имена файлов чекпоинтов номером шага обучения. Чтобы избежать заполнения дисков, сохранители управляют файлами чекпоинтов автоматически. Например, они могут хранить только N последних файлов или один чекпоинт каждые N часов обучения.

Вы нумеруете имена файлов чекпоинтов, передав значение необязательному аргументу global_step в save():

saver.save(sess, 'my-model', global_step=0) ==> filename: 'my-model-0'
...
saver.save(sess, 'my-model', global_step=1000) ==> filename: 'my-model-1000'

Кроме того, необязательные аргументы конструктора Saver() позволяют контролировать распространение файлов чекпоинтов на диске:

• max_to_keep указывает максимальное количество последних файлов чекпоинтов, которые нужно сохранить. По мере создания новых файлов старые удаляются. Если None или 0, чекпоинты не удаляются из файловой системы, а в файле checkpoint сохраняется только последний. По умолчанию значение равно 5 (то есть сохраняются 5 последних файлов чекпоинтов).

• keep_checkpoint_every_n_hours: В дополнение к сохранению последних max_to_keep файлов чекпоинтов, вы можете сохранить один файл чекпоинта каждые N часов обучения. Это может быть полезно, если вы хотите позже проанализировать, как модель развивалась во время длительной сессии обучения. Например, передача keep_checkpoint_every_n_hours=2 гарантирует, что вы сохраняете один файл чекпоинта каждые 2 часа обучения. Значение по умолчанию 10000 часов фактически отключает эту функцию.

Обратите внимание, что для сохранения модели все равно необходимо вызвать метод save(). Передача этих аргументов в конструктор не сохранит переменные автоматически.

Программа обучения, которая регулярно сохраняет данные, выглядит следующим образом:

...
# Create a saver.
saver = tf.compat.v1.train.Saver(...variables...)
# Launch the graph and train, saving the model every 1,000 steps.
sess = tf.compat.v1.Session()
for step in xrange(1000000):
    sess.run(..training_op..)
    if step % 1000 == 0:
        # Append the step number to the checkpoint name:
        saver.save(sess, 'my-model', global_step=step)

В дополнение к файлам чекпоинтов, сохранители хранят протокол буфера на диске со списком последних чекпоинтов. Это используется для управления нумерованными файлами чекпоинтов и latest_checkpoint(), что облегчает поиск пути к последнему чекпоинту. Этот протокол буфера хранится в файле с именем 'checkpoint' рядом с файлами чекпоинтов.

Если вы создаете несколько сохранителей, вы можете указать имя файла протокол буфера в вызове save().

Методы

as_saver_def

Просмотреть исходный код

as_saver_def()

Генерирует представление SaverDef этого сохранителя.

build

Просмотреть исходный код

build()

export_meta_graph

Просмотреть исходный код

export_meta_graph(
    filename=None, collection_list=None, as_text=False, export_scope=None,
    clear_devices=False, clear_extraneous_savers=False, strip_default_attrs=False,
    save_debug_info=False
)

Записывает MetaGraphDef в save_path/filename.

from_proto

Просмотреть исходный код

@staticmethod
from_proto(
    saver_def, import_scope=None
)

Возвращает объект Saver , созданный из saver_def.

recover_last_checkpoints

Просмотреть исходный код

recover_last_checkpoints(
    checkpoint_paths
)

Восстанавливает внутреннее состояние сохранителя после сбоя.

Этот метод полезен для восстановления состояния "self._last_checkpoints".

Поиск чекпоинтов, на которые указывают checkpoint_paths. Если файлы существуют, используйте их mtime в качестве метки времени чекпоинта.

restore

Просмотреть исходный код

restore(
    sess, save_path
)

Восстанавливает ранее сохранённые переменные.

Этот метод выполняет операции, добавленные конструктором для восстановления переменных. Он требует сессии, в которой был запущен граф. Переменные для восстановления не обязательно должны быть инициализированы, так как восстановление само по себе является способом инициализации переменных.

Аргумент save_path обычно представляет значение, ранее возвращённое вызовом save(), или вызовом latest_checkpoint().

save

Просмотреть исходный код

save(
    sess, save_path, global_step=None, latest_filename=None,
    meta_graph_suffix='meta', write_meta_graph=True, write_state=True,
    strip_default_attrs=False, save_debug_info=False
)

Сохраняет переменные.

Этот метод выполняет операции, добавленные конструктором для сохранения переменных. Он требует сессии, в которой был запущен граф. Сохраняемые переменные также должны быть инициализированы.

Метод возвращает префикс пути к вновь созданным файлам контрольных точек. Эту строку можно напрямую передать в вызов restore().

set_last_checkpoints

Просмотреть исходный код

set_last_checkpoints(
    last_checkpoints
)

УСТАРЕВШЕЕ: Используйте set_last_checkpoints_with_time.

Устанавливает список старых имён файлов контрольных точек.

set_last_checkpoints_with_time

Просмотреть исходный код

set_last_checkpoints_with_time(
    last_checkpoints_with_time
)

Устанавливает список старых имён файлов контрольных точек и временных меток.

to_proto

Просмотреть исходный код

to_proto(
    export_scope=None
)

Преобразует этот Saver в протобуфер SaverDef.