https://codefrydev.in/Reinforcement/dl-foundations/training-loop/