ロック画面?スリープじゃなくて?
Wi-Fi詳細設定の「スリープ時にWi-Fi接続を維持」とか「接続不良のとき無効にする」とか「Wi-Fi最適化」とかいじった?