ADB Wi-Fi connection disconnects when device goes to sleep in Moto G 2014
So it seems the problem was with the settings of the device. Having a Screen Lock (even if SmartLock is enabled, and you are in a "safe location") disconnects ADB on device idle. Only way around it for me was to disable the device screen lock.