Don't show grad norm during optimization