Is it possible to save TF model with the whole information around training such as history and at what epoch the training ended? I need this information for the sake of the
