fix a bug when loading model from checkpoint file, rnorm scale