Cg Lai

Thanks @tjruwase for reply. Actually I enable the checkpoint for training. But the OOM problem occurs at the preparation stage. ``` run_clm_no_trainer.py:664 in │ │ │ │ │ │ 661...

Thanks @jomayeri. I will try it.:)