save training loss code example

Example: load model keras

from tensorflow import keras
model = keras.models.load_model('path/to/location')