Pytorch Batch Normalization During The Testing