Direct shift gearbox training