Living arrangements once baby is here, do you think this will work?

Do you think it could work if my partner and I keep both our own places and just some nights stay at his place and some nights stay at mine once baby is here?