Once you get married, you can apply for AP which takes 1-2 months according to my attorney. Once you've got that AP, you are free to go back to China.