前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >python.InvalidArgumentError: Assign requires shapes of both tensors to match.

python.InvalidArgumentError: Assign requires shapes of both tensors to match.

作者头像
平凡的学生族
发布2019-05-25 13:13:54
2.2K0
发布2019-05-25 13:13:54
举报
文章被收录于专栏:后端技术后端技术

问题

代码语言:javascript
复制
INFO:tensorflow:Restoring parameters from /tmp/mnist_convnet_model\model.ckpt-1619
2018-04-10 16:02:17.996575: W c:\l\tensorflow_1501918863922\work\tensorflow-1.2.1\tensorflow\core\framework\op_kernel.cc:1158] Not found: Key conv___1/bias not found in checkpoint
2018-04-10 16:02:17.999870: W c:\l\tensorflow_1501918863922\work\tensorflow-1.2.1\tensorflow\core\framework\op_kernel.cc:1158] Not found: Key conv___1/kernel not found in checkpoint
Traceback (most recent call last):
  File "E:\anaconda\envs\py36\lib\site-packages\tensorflow\python\client\session.py", line 1139, in _do_call
    return fn(*args)
  File "E:\anaconda\envs\py36\lib\site-packages\tensorflow\python\client\session.py", line 1121, in _run_fn
    status, run_metadata)
  File "E:\anaconda\envs\py36\lib\contextlib.py", line 88, in __exit__
    next(self.gen)
  File "E:\anaconda\envs\py36\lib\site-packages\tensorflow\python\framework\errors_impl.py", line 466, in raise_exception_on_not_ok_status
    pywrap_tensorflow.TF_GetCode(status))
tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [64] rhs shape= [32]
         [[Node: save/Assign = Assign[T=DT_FLOAT, _class=["loc:@conv2d/bias"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/cpu:0"](conv2d/bias, save/RestoreV2)]]

解决过程

  1. 查看到INVALIDARGUMENTERROR (SEE ABOVE FOR TRACEBACK): ASSIGN REQUIRES SHAPES OF BOTH TENSORS TO MATCH. LHS SHAPE= [2048,38] RHS SHAPE= [2048,2] 中提到 做tensorflow object detection 中,清空下checkpoint就可以啦
  2. 于是我看到自己的报错信息中有一句Restoring parameters from /tmp/mnist_convnet_model\model.ckpt-1619,说明程序是在读取参数时出错的。尝试删除这些.ckpt-1619后缀的文件,并且不要设置model_dir之类的变量即可。
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2018.04.10 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 问题
  • 解决过程
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档