AnswerDotAI / AnswerDotAI/RAGatouille

trainer.train returns None

Open
#185 1 comment 0 reactions 0 assignees View on GitHub
good first issue help wanted Upstream ColBERT Interaction
Dominant language
Python
Stars
4k
Forks
276
PR merge metrics
No merged PRs in 30d

Description

Hello there,

In the source code, trainer.train is supposed to return the model path, however, it is returning None for me... is there a way to save the trained model in a specific path? or not saving it at all and use it as a variable?

I have not changed anything in the example code

```
model_path = trainer.train(batch_size=32,
nbits=4, # How many bits will the trained model use when compressing indexes
maxsteps=500000, # Maximum steps hard stop
use_ib_negatives=True, # Use in-batch negative to calculate loss
dim=128, # How many dimensions per embedding. 128 is the default and works well.
learning_rate=5e-6, # Learning rate, small values ([3e-6,3e-5] work best if the base model is BERT-like, 5e-6 is often the sweet spot)
doc_maxlen=256, # Maximum document length. Because of how ColBERT works, smaller chunks (128-256) work very well.
use_relu=False, # Disable ReLU -- doesn't improve performance
warmup_steps="auto", # Defaults to 10%
)

```

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by tracing the trainer.train entry point used in the example and follow how training output is saved or returned. Reproduce the call shown in the issue, then verify whether the documented behavior is a model path, an in-memory model, or no return value and update the behavior or documentation accordingly.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.