can't access my finetuned model

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

Start with the RAGTrainer.train call and inspect the returned training_path, then trace how the MyFineTunedColBERT model and checkpoints are written. Reproduce the training flow and determine what is needed to access the finetuned model after training, with a locally reloadable model or checkpoints as the done condition.

Written by the indexing model from the issue text.