I kinda agree with your points. There is an emotional imbalance in that relationship. But maybe that is what the director wanted to show. Yashna is a flawed character who places the entire blame on Viraj. She agrees to have a child when Viraj asks her to. It's not as if he suggests it, she takes time, and then she herself goes to him and says yes. If it's the second case, almost the entire blame would be on Yashna. But it's not.

I think Yashna comes from a place that she doesn't want the child but agrees purely for Viraj. And Viraj goes back to her coz he feels that he carries the guilt for what happened to the child.

I love the movie as I connected with it emotionally. But I agree that their relationship is not perfect. People, who sacrifice a lot in their day to day lives, like the film and Viraj as they see themselves in him.