发生异常: tensorflow.python.framework.errors_impl.InvalidArgumentError
Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error: Assign requires shapes of both tensors to match. lhs shape= [256,3136] rhs shape= [256,5011]
如图:

解决办法:
1. 删去之前训练得到的checkpoint,然后重新训练。同时删除之前生成logs文件夹。
2. 或者可以删除某些checkpoint,让程序可以继续从最高的step开始。

本文解析了在使用TensorFlow框架进行模型训练时遇到的InvalidArgumentError错误,该错误源于模型图与checkpoint不匹配。文章提供了两种解决方案:一是删除原有checkpoint和logs文件夹后重新训练;二是选择性删除checkpoint,使训练能从最后一步继续。
8706





