I am also facing the same issue of wifi getting dropped once the iPhone is on sleep mode. When I unlock it then I am facing no issues with wifi. Please look into this