Dear Mauricio,

Thank you very much for this great explanation. It is beneficial for my research. I have successfully applied your model and extended it into building identification. However, I am having problems saving the checkpoints and the weights. My goal is later to use them to predict or continue training.

I really appreciate it if you could guide me on where and how to implement the function to save the model.

Best regards,

Sebastian