luyug / Condenser

EMNLP 2021 - Pre-training architectures for dense retrieval
Apache License 2.0
243 stars 23 forks source link

How to resume_from_checkpoint #15

Open ToanKGO opened 2 years ago

ToanKGO commented 2 years ago

Hi, I'm newbie in Condenser. I use colab to run code and it can only run 24 hours max. I didn't find any resume_from_checkpoint in your code. How can i continue my training after 24 hours

luyug commented 2 years ago

You should be able to pass your checkpoint directly through the argument --model_name_or_path.

zlh-source commented 2 years ago

Hi, does _num_trainepochs need to be modified? If not, will the entire epoch be run, not the rest of the epoch?