Können Sie Variablen in Tensorflow

Question

Ich habe einen folgenden Code laden:

sess = tf.Session() 
model_file = './runs/1488368349/checkpoints/model-42200.meta' 
checkpoint_dir = './runs/1488368349/checkpoints/' 


new_saver = tf.train.import_meta_graph(model_file) 
#new_saver.restore(sess, MODEL_PATH) 
new_saver.restore(sess, tf.train.latest_checkpoint(checkpoint_dir)) 

all_vars = tf.get_collection('vars') 
print len(all_vars) 
for v in all_vars: 
     v_ = sess.run(v) 
     print(v_) 

Aber der obige Code kann nicht mehr Variablen laden. Ich bekomme die Länge von all_vars als 0.
Kann mir jemand sagen, was das Problem ist?

Answer 1

Können Sie das versuchen?

tf.get_collection(tf.GraphKeys.VARIABLES) 

Dies sollten Sie alle trainierbar und untrainable Variablen geben ...