https://camera-optical-zoom.blogspot.com/2024/07/saving-custom-estimators-in-tensorflow.html