I usually just stay stroking DS if he's awake, unless he's amused & playing.
If he's busy playing he usually puts himself to sleep, but if he wakes when I put him in I'll stroke & pat him till he's asleep.
I think its easier & quicker to stay, but its up to you & whatever works for you.