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 os
f
原创
2020-09-10 14:15:21 ·
524 阅读 ·
0 评论