England selector James Whitaker has revealed that all-rounder Ben Stokes will return to training next week as he steps up his recovery from a wrist injury.