tensorflow保存读取数据出现NotFoundError (see above for traceback): Key v1_3 not found in checkpoint [[Node

我就是我 2022-02-01 05:19 179阅读 0赞

在保存模型的最后面添加：`tf.reset_default_graph()`  
在读取模型中添加：`tf.get_variable_scope().reuse_variables()`

### 原始代码： ###

# 读取模型
    import tensorflow as tf
    import numpy as np
    
    ## Save to file
    # remember to define the same dtype and shape when restore
    W = tf.Variable([[1,2,3],[3,4,5]], dtype=tf.float32, name='weights')
    b = tf.Variable([[1,2,3]], dtype=tf.float32, name='biases')
    
    # init= tf.initialize_all_variables() # tf 马上就要废弃这种写法
    # 替换成下面的写法:
    init = tf.global_variables_initializer()
    saver = tf.train.Saver()
    
    with tf.Session() as sess:
        sess.run(init)
        save_path = saver.save(sess, "my_net/save_net.ckpt")
        print("Save to path: ", save_path)

# 加载模型
    # 先建立 W, b 的容器
    W = tf.Variable(np.arange(6).reshape((2, 3)), dtype=tf.float32, name="weights")
    b = tf.Variable(np.arange(3).reshape((1, 3)), dtype=tf.float32, name="biases")
    
    # 这里不需要初始化步骤 init= tf.initialize_all_variables()
    
    saver = tf.train.Saver()
    with tf.Session() as sess:
        # 提取变量
        saver.restore(sess, "my_net/save_net.ckpt")
        print("weights:", sess.run(W))
        print("biases:", sess.run(b))

报错：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4NDEwNDI4_size_16_color_FFFFFF_t_70]

### 修改后代码 ###

# 加载模型
    # 先建立 W, b 的容器
    W = tf.Variable(np.arange(6).reshape((2, 3)), dtype=tf.float32, name="weights")
    b = tf.Variable(np.arange(3).reshape((1, 3)), dtype=tf.float32, name="biases")
    
    # 这里不需要初始化步骤 init= tf.initialize_all_variables()
    
    saver = tf.train.Saver()
    with tf.Session() as sess:
        # 提取变量
        tf.get_variable_scope().reuse_variables() #添加的代码
        saver.restore(sess, "my_net/save_net.ckpt")
        print("weights:", sess.run(W))
        print("biases:", sess.run(b))

运行结果：  
![在这里插入图片描述][2019051315484938.png]

### 原因： ###

首先查看变量的dtype和shape  
1、首先是自己定义的参数变量是否和保存的参数变量的类型是一致的

2、在文件下面是否有一个叫做chockpoint的东西存在

3、最后如果运行多次出现NotFoundError (see above for traceback): Key Variable\_4 not found in checkpoint

\[\[Node: save\_2/RestoreV2 = RestoreV2\[dtypes=\[DT\_FLOAT, DT\_FLOAT, DT\_FLOAT,

这种情况，请使用tf.reset\_default\_graph()，还有就是这句话要放在前面一点。

4、还有就是保存好了模型，如果使用的是spyder请先关闭，之后在尝试读入模型。

#### 如果以上都无法解决你的问题，请删除模型之后，重新写入，再载入试试。 ####

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4NDEwNDI4_size_16_color_FFFFFF_t_70]: /images/20220201/bedc2551df6942e8bc3fe08c4126d4c2.png
[2019051315484938.png]: /images/20220201/648828e67a6d4cfd8d25ee422d11af96.png