https://travelebookdownload.blogspot.com/2024/08/how-to-load-layer-from-checkpoint.html