fix a bug of can't resume from checkpoint in finetune_clm_lora.py#354