Travel plan to india while the H1B extension is in progress with valid i-94

Hello,

My Visa is getting expired in July 31st 2017 and i will be starting my visa extension process in between . I have a plan to travel India in May last week and will be returning in Mid of June. My question is can i travel and return back to US during the visa extension is in process. If not how soon i need to start the process ?

Can some one please assist.