I am trying to replicate the loss reported after training a tf.keras model. From my understanding, hist