Keras
还是我666
这个作者很懒,什么都没留下…
展开
-
【代码报错】tensorflow.python.framework.errors_impl.InvalidArgumentError: Dimensions must be equal
报错截图 问题出自于我的custom layer。Keras在初始化custom layer的时候,必须加上compute_output_shape函数才能自动判断input_shape。 否则的话,如果实例化了两个custom layer: L_1和L_2。在不加上述函数的情况下,L_2层的input_shape会默认和L_1层相同。 在keras自己的文档上其实说的很清楚了,但是比较容易忽视。以下是链接 https://keras.io/zh/layers/writing-your-own-..原创 2020-05-18 06:42:23 · 1193 阅读 · 0 评论 -
代码报错:ValueError: An operation has `None` for gradient. Please make sure that all of your ops have a
自定义Keras层后,代码报错,提示如标题,无法计算gradient。 原因:我是在别人模型的基础上加上自己的代码,修改了代码之后,原来作者使用add_weight初始化的变量矩阵没有删除。导致错误。 如图,没有用到的变量需要及时删除,否则在计算gradient时会出错 ...原创 2020-04-09 13:29:26 · 4978 阅读 · 0 评论