【发布时间】:2019-05-28 22:53:53
【问题描述】:
我将 Keras/TF 与以下模型一起使用:
conv = Conv2D(4, 3, activation = None, use_bias=True)(inputs)
conv = Conv2D(2, 1, activation = None, use_bias=True)(conv)
model = Model(input = inputs, output = conv)
model.compile(optimizer=Adam(lr=1e-4), loss=keras.losses.mean_absolute_error)
在 model.fit 中,我收到一条错误消息:
ValueError:检查目标时出错:预期 conv2d_2 有 形状 (300, 320, 2) 但得到了形状 (300, 320, 1) 的数组
这符合预期,因为目标是单通道图像,而模型中的最后一层有 2 个通道。
我不明白为什么当我使用自定义损失函数时:
def my_loss2(y_true, y_pred):
return keras.losses.mean_absolute_error(y_true, y_pred)
并编译模型:
model.compile(optimizer = Adam(lr=1e-4), loss=my_loss2)
它确实工作(或者至少没有给出错误)。是否正在进行任何类型的自动转换/截断?
我正在使用 TF (CPU) 1.12.0 和 Keras 2.2.2
真诚地, 埃拉德
【问题讨论】:
-
这很有趣,你确定除了损失函数之外没有任何其他改变,因为在损失周围添加一个包装器,不应该真正改变任何东西。
-
您使用的是哪个 keras 版本?对于 keras 2.2.2,我无法确认这一点。
model.compile不会给我任何损失的错误。但是,当我调用model.fit时,我确实得到了他们俩的错误。实际上,在您发布的代码中,model.compile没有您的目标形状的概念,那么它应该如何检测该错误? -
对不起,我没有在帖子中说明这一点。错误当然发生在 model.fit 中(我会更新帖子)。我使用的是 keras 2.2.2 和 tensorflow (cpu) 1.12.0,哪个版本给了你一致的行为?
-
@Marumba 我刚刚意识到自定义丢失的错误是不同的。发生这种情况是因为我缺少的倒数第二个维度的不一致。如果我解决了这个问题并且所有尺寸都相同,除了最后一个,
fit确实通过了,没有任何自定义丢失错误。显然,最后一个维度正在进行一些广播,但这并不能解释为什么内置损失没有广播......我假设内置损失fit正在做一些手动形状检查...
标签: tensorflow keras loss-function