TensorFlow: Saver hat 5 Modelle zu begrenzen

8

Ich wollte mehrere Modelle für mein Experiment speichern, aber mir ist aufgefallen, dass tf.train.Saver() constructor nicht mehr als 5 Modelle speichern konnte. Hier ist ein einfacher Code:

%Vor%

Als ich diesen Code ausführte, sah ich nur fünf Modelle auf meinem Desktop. Warum ist das? Wie kann ich mehr als 5 Modelle mit demselben tf.train.Saver() -Konstruktor speichern?

    
Transcendental 08.08.2016, 19:44
quelle

1 Antwort

15

Der tf.train.Saver() -Konstruktor verwendet ein optionales Argument genannt max_to_keep , die standardmäßig die 5 letzten Checkpoints Ihres Modells speichert. Um weitere Modelle zu speichern, geben Sie einfach einen Wert für dieses Argument an:

%Vor%

Um alle Prüfpunkte zu behalten, übergeben Sie das Argument max_to_keep=None an den Sparer-Konstruktor.

    
mrry 08.08.2016, 19:49
quelle