2016-09-21 2 views
4

Ich lade von einem gespeicherten Modell und ich möchte in der Lage sein, einen Tensorflow-Optimierer wie Adam Optimizer zurücksetzen. Im Idealfall so etwas wie:Reset Tensorflow Optimizer

sess.run([tf.initialize_variables(Adamopt)]) 

oder

sess.run([Adamopt.reset]) 

ich versucht habe, nach einer Antwort suchen, haben aber noch keine Art und Weise zu finden, es zu tun. Hier ist, was ich gefunden habe, welche Adresse das Problem nicht: https://github.com/tensorflow/tensorflow/issues/634

In TensorFlow is there any way to just initialize uninitialised variables?

Tensorflow: Using Adam optimizer

ich im Grunde nur eine Art und Weise wollen die „slot“ Variablen in der Adam Optimizer zurücksetzen.

Dank

Antwort

1

Der einfachste Weg, ich war dem Optimierungsprogramm einen eigenen Variable Umfang und dann

optimizer_scope = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, 
           "scope/prefix/for/optimizer") 
sess.run(tf.initialize_variables(optimizer_scope)) 

Idee von freeze weights

+0

Das funktioniert nicht mit Adam in TensorFlow 1.0 erscheinen zu arbeiten laufen geben gefunden . – sunside

+0

Ich testete es in tf 0.10 Ich werde den Test mit tf 1.0 wiederholen. Was war der Fehler? – Steven

+0

Sollte angegeben haben - es enthält keine trainierbaren Variablen für Adam, der Bereich scheint leer zu sein. – sunside

Verwandte Themen