Tensorflow, отсутствующие файлы контрольных точек. Сохраняет ли только сохранение 5 контрольных точек?

Question

Я работаю с тензорным потоком и тренировал некоторые модели и сохранял их после каждой эпохи, используя метод tf.saver(). Я могу сохранить и загрузить модели просто отлично, и я делаю это обычным способом.

with tf.Graph().as_default(), tf.Session() as session: 
    initialiser = tf.random_normal_initializer(config.mean, config.std) 

    with tf.variable_scope("model",reuse=None, initializer=initialiser): 
     m = a2p(session, config, training=True) 

    saver = tf.train.Saver() 
    ckpt = tf.train.get_checkpoint_state(model_dir) 
    if ckpt and tf.gfile.Exists(ckpt.model_checkpoint_path) 
     saver.restore(session, ckpt.model_checkpoint_path) 
    ... 
    for i in range(epochs): 
     runepoch() 
     save_path = saver.save(session, '%s.ckpt'%i) 

Мой код настроен для сохранения модели для каждой эпохи, которая должна быть маркирована соответствующим образом. Однако я заметил, что после пятнадцати эпох обучения у меня есть только файлы контрольных точек за последние пять эпох (10, 11, 12, 13, 14). В документации ничего не говорится об этом, поэтому я не понимаю, почему это происходит.

Сохраняется ли только сохранение пяти контрольно-пропускных пунктов или я сделал что-то неправильно?

Есть ли способ убедиться, что все контрольно-пропускные пункты сохранены?

Answer 1

Вы можете выбрать, сколько контрольных точек, чтобы сохранить, когда вы create your Saver object, установив max_to_keep аргумент, который по умолчанию 5.

saver = tf.train.Saver(max_to_keep=10000) 

Answer 2

установка max_to_keep=None фактически делает Saver держать все контрольно-пропускные пункты. Для например.,

saver = tf.train.Saver(max_to_keep=None)