Key lstm/BW/basic_lstm_cell/bias not found in checkpoint

最新推荐文章于 2021-05-07 20:29:25 发布

lydiayl

最新推荐文章于 2021-05-07 20:29:25 发布

阅读量521

点赞数

分类专栏： TensorFlow 文章标签： tensorflow

本文链接：https://blog.csdn.net/weixin_44393192/article/details/108513005

版权

TensorFlow 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

Key lstm/BW/basic_lstm_cell/bias not found in checkpoint

问题
今天恢复模型的时候一直报这个错误： Key lstm/BW/basic_lstm_cell/bias not found in checkpoint。查了一下资料，发现大都是说tf版本不同的问题。但奇怪的是，我自己是在 tf1.14的环境中训练的模型，按理再从 tf1.14恢复不应该出现这个问题才是。

解决办法
参考这个作者的一个代码
打印出自己ckpt文件的张量名字。

import os
from tensorflow.python import pywrap_tensorflow
reader = pywrap_tensorflow.NewCheckpointReader('../model/model.ckpt-100000') #替换为自己的路径
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
    print("tensor_name: ", key)

结果如下
在这里插入图片描述
确实有这个变量呀，只不过bw是小写。然后把代码中的BW改成小写bw就行了。。

lydiayl

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
Key lstm/BW/basic_lstm_cell/bias not found in checkpoint

Key lstm/BW/basic_lstm_cell/bias not found in checkpoint问题今天恢复模型的时候一直报这个错误：Key lstm/BW/basic_lstm_cell/bias not found in checkpoint。查了一下资料，发现大都是说tf版本不同的问题。但奇怪的是，我自己是在tf1.14的环境中训练的模型，按理再从tf1.14恢复不应该出现这个问题才是。解决办法参考这个作者的一个代码打印出自己ckpt文件的张量名字。import osf
复制链接

扫一扫