You may try turn off power-saving in that particular wifi connection.

Go setting and choose internet connect, edit that connection, press Next until you see Advance button, the power-saving setting is under Other.

I've to turn off power-saving in my home wifi connection so that my ssh session could always stay on. YMMV.

Hope this help.