Tensorflow has two separate functions to calculate MSE (Mean square error). For Loss - tf.keras.loss.MeanSquaredError() For Metrics - tf.keras.metrics.MeanSquaredError()