Ayuei / DeBEIR

Dense Bi-Encoder Retrieval for Rapid Experimentation
GNU General Public License v3.0
5 stars 0 forks source link

errors with trec2022 examples #12

Closed KonradHoeffner closed 1 year ago

KonradHoeffner commented 1 year ago
(venv) trec2022$ python training.py             
Traceback (most recent call last):
  File "/home/konrad/tmp/debeir/examples/trec2022/training.py", line 11, in <module>
    from training.utils import DatasetToSentTrans
  File "/home/konrad/tmp/debeir/examples/trec2022/training.py", line 11, in <module>
    from training.utils import DatasetToSentTrans
ModuleNotFoundError: No module named 'training.utils'; 'training' is not a package
(venv) trec2022$ python train.py   
Traceback (most recent call last):
  File "/home/konrad/tmp/debeir/examples/trec2022/train.py", line 16, in <module>
    from training.utils import get_scheduler_with_wandb
  File "/home/konrad/tmp/debeir/examples/trec2022/training.py", line 11, in <module>
    from training.utils import DatasetToSentTrans
ModuleNotFoundError: No module named 'training.utils'; 'training' is not a package
Ayuei commented 1 year ago

Fixed in https://github.com/Ayuei/DeBEIR/commit/e0047b75de10c9ef40c47bfee01568fb0a7a92ef Newer examples are less specific about the task and use the updated API.

KonradHoeffner commented 1 year ago

As the folder trec2022 does not exist anymore I assume it got renamed to finetune_biencoder? That one now successfully starts training on my machine, however I cancelled it as it showed 4 hours of expected runtime, but it looks like it works correctly.