train.py 文件源码

def restore_variables(sess, saver, pretrained_model):
    saver.restore(sess, pretrained_model)
    uninit_vars = []
    for var in tf.all_variables():
        try:
            sess.run(var)
        except tf.errors.FailedPreconditionError:
            uninit_vars.append(var)

    init_new_vars_op = tf.initialize_variables(uninit_vars)
    sess.run(init_new_vars_op)