错误信息: ValueError: Shape must be rank 4 but is rank 0 for ‘Conv2D’ (op: ‘Conv2D’) with input shape:[32, 299, 299, 3], []
错误原因:由于网络实现过程中权重初始化函数只写了接口,但是没有实现,导致卷积没有权重。 解决方法:为卷积运算添加权重后问题解决
如果错误信息中指示第二个shape不是空的,比如:input shape: [32, 299, 299, 3], [45, 5, 90, 88], 这时候应该检查一下输入的shape是不是有问题
错误信息:ValueError: Rank mismatch: Rank of labels (received 2) should equal rank of logits minus 1 (received 2)
错误原因:传入的label的rank错误,应该是batch size大小的一维向量,比如:模型输出是batch size是32的1000分类的结果,则输出shape为[32, 1000],这时传入交叉熵函数的label应该是一个batch 32张图片的label shape为[32]
解决办法:修改构造label的shape