Open GP24CairoUni opened 5 months ago
from ragatouille import RAGTrainer
trainer = RAGTrainer(model_name="MyFineTunedColBERT", pretrained_model_name="colbert-ir/colbertv2.0")
trainer.prepare_training_data(raw_data=positive_pairs, data_out_path="./data2/", all_documents=cleaned_documents, mine_hard_negatives=True)
training_path = trainer.train(batch_size=16, nbits=4, maxsteps=500000, use_ib_negatives=True, dim=128, learning_rate=5e-6, doc_maxlen=256, use_relu=False, warmup_steps="auto")
I finished finetuning but I cant access and use the model again and no checkpoints on my local storage
may be try to export it into the huggingface hub and then download it again, but didn't find it in the path is extremely weird.
from ragatouille import RAGTrainer
trainer = RAGTrainer(model_name="MyFineTunedColBERT", pretrained_model_name="colbert-ir/colbertv2.0")
trainer.prepare_training_data(raw_data=positive_pairs, data_out_path="./data2/", all_documents=cleaned_documents, mine_hard_negatives=True)
training_path = trainer.train(batch_size=16, nbits=4, maxsteps=500000, use_ib_negatives=True, dim=128, learning_rate=5e-6, doc_maxlen=256, use_relu=False, warmup_steps="auto")
I finished finetuning but I cant access and use the model again and no checkpoints on my local storage