How to build a custom production-ready Deep Learning Training loop in ...