If you would actually like to write the music down (as in sheet music), use Finale NotePad or actually buy plain sheets of music and write it down.

The only other way is to have an amazing memory.

I usually just play around on my guitar until I play something and think it would sound good, and build the song around that. If I enjoy playing the song I always remember how to do it so I don't have the forgetting problem.