tensorflow.python.framework.errors_impl.InvalidArgumentError

tensorflow.python.framework.errors_impl.InvalidArgumentError:

logits and labels must be broadcastable: logits_size=[8,4] labels_size=[8,2]
训练模型时报错,tensorflow.python.framework.errors_impl.InvalidArgumentError: logits and labels must be broadcastable: logits_size=[8,4] labels_size=[8,2]。在网上查阅资料有的说是显存不足需要减小batchsize等参数,也有的说是图像数据的格式不匹配。但这个错误没能通过以上方式解决,最后发现输出层的输出个数与数据集的种类个数不一致所导致的报错,只要修改输出层的输出个数即可。

仅供参考

猜你喜欢

转载自blog.csdn.net/weixin_46178278/article/details/109502846
今日推荐