Tensorflow错误:logits和标签必须大小相同:logits_size = [768,64] labels_size = [48,64]

时间:2018-03-23 10:25:13

标签: python-3.x tensorflow conv-neural-network

我正在尝试构建一个简单的(没有上采样)完全卷积神经网络。但是,我遇到了以下错误:

InvalidArgumentError (see above for traceback): logits and labels must be same size: logits_size=[768,64] labels_size=[48,64]
 [[Node: SoftmaxCrossEntropyWithLogits_2 = SoftmaxCrossEntropyWithLogits[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/cpu:0"](Reshape_9, Reshape_10)]]

很明显,这个错误的原因是logits和标签有不同的大小,但我完全不知道代码的哪一部分可能对此负责。给feed_dict的数据集似乎没问题。

以下是代码:https://gist.github.com/den250400/1641d20dc31fbcf1070211451eac7628

0 个答案:

没有答案