以下是我在喀拉拉邦的自定义损失函数。
y_true
和y_pred
的形状均为(256,384,3)
。
def myloss(y_true,y_pred):
diff = y_pred - y_true
diff = diff[96:160 , 144:240]
return K.mean(K.square(diff), axis=-1)
在这里,我将部分差异切成薄片,并为此找到mean_square_loss
。但是,此函数返回nan
作为损失。为什么会这样?