starcoder icon indicating copy to clipboard operation
starcoder copied to clipboard

Is it possible to continue training from the last lora checkpoint?

Open edwardelric1202 opened this issue 2 years ago • 1 comments

Hi, I wonder that if I can continue finetune this model from the last lora checkpoint if I interrupted training process?

edwardelric1202 avatar May 10 '23 08:05 edwardelric1202

With PyTorch trainer you can replace trainer.train()

with trainer.train(resume_from_checkpoint=<PATH>) e.g. "checkpoint-1234" or trainer.train(resume_from_checkpoint = True)

More info from HF docs

brainwavecoder9 avatar Jun 07 '23 06:06 brainwavecoder9