Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
157 views
in Technique[技术] by (71.8m points)

python - Recover Tensorflow variables from model

I have been working with a TF model and I want to restore the model variables. However, these variables were never saved during the Training stage, and I'm wondering if there is a way to recover those from just the model I recreated from just the .h5 weights file.

This is how I'm recovering the model:

import tensorflow as tf

with tf.Session(config=conf) as sess:
    restored_model = my_custom_model(sess, ...)
    restored_model.load_weights('/path/to/file.h5')

The my_custom_model() definition initializes variables like this:

    # Init vars before training
    sess.run(tf.local_variables_initializer())
    sess.run(tf.global_variables_initializer())
    sess.run(tf.tables_initializer())
    K.set_session(sess)

However, during training, the model variables were not saved. Is there a way to get these variables?


与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)
等待大神答复

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...