tensorflow 连续restore时碰见的bug以及解决方法加载tensorboard可视化

最新推荐文章于 2020-11-25 21:37:31 发布

转载最新推荐文章于 2020-11-25 21:37:31 发布 · 1.8k 阅读

本文介绍了一种常见的模型权重加载错误“NotFoundError: KeyVariable_2 not found in checkpoint”，该问题通常发生在尝试恢复模型时发现权重名称已更改。解决方法是在恢复操作前加入tf.reset_default_graph()，以确保每次恢复时模型都会重置为默认图。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

NotFoundError: Key Variable_2 not found in checkpoint
	 [[Node: save_1/RestoreV2 = RestoreV2[dtypes=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save_1/Const_0_0, save_1/RestoreV2/tensor_names, save_1/RestoreV2/shape_and_slices)]]

解决方法:

搜了半天，是因为再次读取模型时，权重名称已经发生变化，所以就会报错误。详情请参考这篇博客。

解决方法：在resume前面加上tf.reset_default_graph()，这样使得每次恢复时模型会转化成默认的图。

tensorboard --logdir=G:\tensorflow\