- 博客(2)
- 收藏
- 关注
原创 【代码报错】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
1229
原创 代码报错: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
5005
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人