Replace as well the regression loss with Keras loss object. PiperOrigin-RevId: 316062800
Attach a file by drag & drop or click to upload
