Thank you, Vahid!
I’m interested in finding the variance of  when training a model across different datasets at a fixed parameter value , i.e., the variance of gradient estimator. So I think I need to stop after training the model a few epochs to calculate the gradient and do it multiple times to get the variance, can you suggest a way to do that?
Thanks