InvalidArgumentError:

Inputs to operation train_op/gradients/AddN of type _MklAddN must have the same size and shape. 

Input 0: [20,8,8,128] != input 1: [163840]

出现上面的报错,有可能是python3.7版本问题,可以改为python3.6。

解决来源参考:https://github.com/zhixuhao/unet/issues/96

另外在anaconda上另外安装python3.6时,可能会碰到报错:RemoveError: ‘setuptools’ is a dependency of conda

解决办法可参考我的博客:https://www.jianshu.com/p/3ef3154417e4

文章来源于互联网,如有雷同请联系站长删除:用TensorFlow跑ResNet卷积神经网络,报错InvalidArgumentError

发表评论