DeepSpeed

call empty_cache to really free up GPU memory as described in comment

#2620

Merged