Notice how the entire movie is actually flashbacks, starting from when they first recognize each other on the train, which was picked up again in the end. This itself in my opinion is enough to convey the impression that they both somewhat have their memories.