Remove data after training to free up memory.