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