Onset of non-linear internal gravity waves in intermediate-mass stars