Open salbatarni opened 6 months ago
Hey, it returning None is a quirk/bug of the upstream ColBERT implementation. The model is by default saved to disk and there's currently no way to disable this behaviour (though it'd be a good change in the future!).
Passing a custom path is also slightly complex, you'd need to update both the root_path and experiments_path in the ColBERT config, which is again something that could be simplified, either here or upstream.
Marking this as a good first issue and leaving it open as those would be welcome QoL changes, especially as I'm starting to work on improving training!
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