machine-learning - TensorFlow错误: "logits and labels must be same size", WarmSpringWinds "tutorial"

Question

我目前正在关注此 tutorial由于 tensorflow 更新而进行了一些更改后，我收到了此错误:

tensorflow.python.framework.errors_impl.InvalidArgumentError: logits and labels must be same size: logits_size=[399360,2] labels_size=[409920,2] [[Node: SoftmaxCrossEntropyWithLogits = SoftmaxCrossEntropyWithLogits[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/cpu:0"](Reshape_2, Reshape_3)]].

谁能帮我解决这个问题吗？

代码中的更改:

#Replaced concat_dim=2 with axis=2
combined_mask = tf.concat(axis=2, values=[bit_mask_class, bit_mask_background])

#Update the import of urllib2 to urllib3

#Replace tf.pack with tf.stack
upsampled_logits_shape = tf.stack([
downsampled_logits_shape[0],
downsampled_logits_shape[1] * upsample_factor,
downsampled_logits_shape[2] * upsample_factor,
downsampled_logits_shape[3]])

Answer 1

引发错误的原因是，当您向函数提供 409920 标签时，logits 数为 399360。函数 tf.nn.softmax_cross_entropy_with_logits 期望每个 logit 都有一个标签，但它会崩溃，因为您提供的标签多于 logits。

至于为什么会发生这种情况，您应该发布对代码所做的更改。