We do not currently know of any antonyms for lose one's train of thought.
For similar words to lose one's train of thought, try: