I'm having the exact same thing. Sometimes I'll just stay awake for an hour counting breaths into the 1000s, other times I'll drift and fall asleep. How do I get that right amount of awareness that keeps me focused but not so "in the zone" enough to stay awake for hours?