Функция latest_checkpoint

Функция latest_checkpoint возвращает полный путь к последнему чекпоинту, сохраненному в заданной директории. Первым параметром функция принимает строку с путем к директории, в которой хранятся файлы чекпоинтов. Вторым необязательным параметром можно передать префикс имени файла чекпоинта. Если чекпоинты не найдены, функция возвращает значение None.

Синтаксис

tf.train.latest_checkpoint(checkpoint_dir, [latest_filename])

Пример

Давайте создадим директорию, сохраним в нее чекпоинт и найдем путь к последнему чекпоинту:

import tensorflow as tf
import os

checkpoint_dir = '/tmp/tf_checkpoints'
os.makedirs(checkpoint_dir, exist_ok=True)

t = tf.constant([1, 2, 3, 4, 5])
ckpt = tf.train.Checkpoint(var=t)
ckpt.write(os.path.join(checkpoint_dir, 'model.ckpt'))

res = tf.train.latest_checkpoint(checkpoint_dir)
print(res)

Результат выполнения кода:

"/tmp/tf_checkpoints/model.ckpt"

Пример

Давайте сохраним два чекпоинта и убедимся, что функция возвращает путь к последнему из них:

import tensorflow as tf
import os

checkpoint_dir = '/tmp/tf_checkpoints2'
os.makedirs(checkpoint_dir, exist_ok=True)

t = tf.constant([1, 2, 3, 4, 5])
ckpt = tf.train.Checkpoint(var=t)

ckpt.write(os.path.join(checkpoint_dir, 'model.ckpt-1'))
ckpt.write(os.path.join(checkpoint_dir, 'model.ckpt-2'))

res = tf.train.latest_checkpoint(checkpoint_dir)
print(res)

Результат выполнения кода:

"/tmp/tf_checkpoints2/model.ckpt-2"

Пример

Давайте проверим поведение функции, когда в директории нет ни одного чекпоинта:

import tensorflow as tf
import os

checkpoint_dir = '/tmp/tf_checkpoints_empty'
os.makedirs(checkpoint_dir, exist_ok=True)

res = tf.train.latest_checkpoint(checkpoint_dir)
print(res)

Результат выполнения кода:

None

Смотрите также

функцию saved_model.save,
которая сохраняет модель в формате SavedModel
функцию saved_model.load,
которая загружает модель из формата SavedModel
класс SaveOptions,
который задает параметры сохранения чекпоинтов
класс LoadOptions,
который задает параметры загрузки чекпоинтов