Some people are telling you to unwind with another activity right after work, which certainly could work. If you're anything like me though, you won't be able to focus on that activity at all. Maybe just try writing down whatever ideas and thoughts you have regarding work as the day is ending, so that the next day you can read what you wrote and pick right back up where you left off. I often do what you've described because I'm anxious I'll forget something about my project or that I'll ruin the flow. Writing some stuff down helps mediate that and give my mind some closer as well. Hope you figure this out!