correction of the squared gradient norm's computation.