tensorflow 运行error

最新推荐文章于 2022-01-19 10:49:16 发布

原创最新推荐文章于 2022-01-19 10:49:16 发布 · 335 阅读

0 ·

CC 4.0 BY-SA版权

本文详细解析了在深度学习模型训练过程中遇到的错误：checkpoint文件中的模型结构与当前运行程序中的模型结构不匹配的问题。该问题表现为在尝试加载checkpoint文件时出现InvalidArgumentError错误，错误信息显示为'Assign requires shapes of both tensors to match'。文章提供了详细的解决方案，即删除现有的checkpoint文件并重新开始训练，以确保模型结构的一致性。

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

ERROR:InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [3,3,3,128] rhs shape= [3,3,1,128] [[Node: save/Assign_36 = Assign[T=DT_FLOAT, _class=["loc:@CNN/c1_W"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](CNN/CNN/c1_W/Adam, save/RestoreV2/_73)]]

原因：checkpoint文件中保存的结构跟新执行程序中数据的结果不匹配。

解决方法：删除checkpoint 文件重新训练