is your response to alon?
btw related: python 3.x - Reset parameters of a neural network in pytorch - Stack Overflow