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

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\

你可能感兴趣的:(tensorflow 连续restore时碰见的bug以及解决方法 加载tensorboard可视化)