Checkpoint callbacks and checkpoint last model

[michele-milesi]

We want to add Fabric callbacks during training checkpoints to improve code readability. Furthermore, we want to add the checkpoint of the model once the training is ended.