R中顺序Keras模型中的自定义损失函数返回nan

Question

我正在尝试在keras中为R的顺序模型中创建自定义函数。用于编译的代码如下：

AE_model %>% compile(
   loss = custom_metric("name",metric_fn= function(y_true, y_pred){
   diff = y_true - y_pred
   sqrt(abs(diff))
}), 
optimizer = optimizer_adam()
)

但是训练是难的时候退回的损失。

但是当我尝试sqrt(0)时，R给了我0。

编辑

我尝试了sgd而不是adam优化器，在第三个时代我再次感到难过。因此，这一定是其他问题，而不是优化程序。