I have a pretrained model, and want to apply it to a new dataset. From the AdaBN paper, I want to recaculate the the mean and var for each neuros and update the value for each BN layer. How could I reset the parameter?