With this departure you are still on IFR flight plan and it's an IFR procedure, so 400ft seems reasonable.
No benefit of turning earlier (especially versus the risks), and in case of flat terrain and no obstacles, no issue that would require flying straight for longer. And depending on the aircraft's AFDS configuration, you might not be able to change the AFDS roll mode below 400ft as well.
Keep it simple.