PyPI page
Home page
Author:
Kexin Wang
License:
Summary:
A package for training and doing inference with contrastive learning with multiple GPUs (Pytorch-DDP).
Latest version:
0.0.8
Required dependencies:
accelerate
|
datasets
|
faiss-cpu
|
more-itertools
|
pytrec-eval
|
sentence-transformers
|
ujson
|
wandb
Downloads last day:
7
Downloads last week:
15
Downloads last month:
71