Solved by mmg in post #7 
              
              
                Yes. The model’s weight get update after each epoch of the training_loop function. Its values get updated. So the validate function will automatically see those updated values when you pass model as an argument