Luton captain Tom Lockyer has returned to training after the cardiac arrest that left him fighting for his life.