Save training loss history

[Gollini]

Hello, I would like to save the training loss history to see the progress using different optimizers.
I am not familiar with TensorFlow. How can I extract this information?
Kind Regards.

[OrsonXu]

Hi Gollini, you may want to use add another callback function to access the loss from the model. Please check an example here: https://stackoverflow.com/questions/65302030/access-loss-and-model-in-a-custom-callback

You can consider adding a callback to the

GLOBEM/algorithm/dl_erm.py

Line 71 in 6b7581d

def prep_callbacks(self, X):