- 博客(2)
- 收藏
- 关注
RSS订阅原创 【代码报错】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
1190
原创 代码报错：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
4967
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人