Thank you once again.
So, I have to do create weight,bias, running_mean and running_var for each layer in init, right?
In your code, I am guessing there’s no storing is done, can you please inform where exactly and how to store?