https://mingredbestcheapprice.blogspot.com/2022/11/saving-custom-estimators-in-tensorflow.html