You can adjust the Wi-Fi sleep policy.
1. Go to Settings ➔ Wi-Fi. 2. Tap the Menu key 3. Tap Advanced. 4. Tap Keep Wi-Fi on during sleep. 5. Select an option.