different calculation of gradnorm, random initialization of weights