2016-12-07 5 views
2

Я использую tensorflow контрольных точки через каждые 10 эпох, используя следующий код:модели Tensorflow проходной получать удаленные

checkpoint_dir = os.path.abspath(os.path.join(out_dir, "checkpoints")) 
checkpoint_prefix = os.path.join(checkpoint_dir, "model") 
... 
if current_step % checkpoint_every == 0: 
    path = saver.save(sess, checkpoint_prefix, global_step=current_step) 
    print("Saved model checkpoint to {}\n".format(path)) 

Проблема в том, что, поскольку новые файлы становятся генерироваться, предыдущие 5 файлы модель становится удалены автоматически ,

ответ

6

Это ожидаемое поведение, документы для tf.train.Saver говорят, что по умолчанию сохраняются 5 последних файлов контрольных точек. Чтобы настроить это, установите max_to_keep нужное значение.

+0

Спасибо @Gregory. Я пропустил эту часть. – Nitin

Смежные вопросы