Impostazioni -> wifi -> avanzate wifi
Selezioni si sotto la voce "mantieni attivo il Wi-Fi durante lo sleep".
Se non funziona prova a riavviare il cellulare e se ancora non funziona potrebbe esserci qualche app per il risparmio energetico che lo fa in automatico, se non ne ha dovresti provare con un reset.