add check gradient norm