2017-02-25 4 views
0

Ich benutze tf.train.Supervisor, um eine Sitzung zu starten, um mein Modell zu trainieren, und Modellparameter alle 1000 Schritte zu speichern. Aber es scheint, dass Tensorflow automatisch alte Modelldateien löscht. Nur 5 aktuelle Modelle wurden gespeichert. Sobald model.ckpt-6000 erzeugt wird, wird das Modell.ckpt-1000 gelöscht. Aber ich kann keine Dokumente zu dieser Operation finden.Tensorflow löscht automatisch alte Modelldateien

Antwort

1

Hier sind die docs: https://www.tensorflow.org/api_docs/python/tf/train/Saver

Es hat eine Variable mit dem Namen max_to_keep, die standardmäßig auf 5. So rufen Sie einfach

tf.train.Saver(max_to_keep=100000) 

oder eine andere große Zahl, die alle umfasst, die erstellt werden könnten. Keine Möglichkeit, alle Modelle als eine Option zu speichern, die ich kenne.

Verwandte Themen