shibing624 / text2vec

text2vec, text to vector. 文本向量表征工具,把文本转化为向量矩阵,实现了Word2Vec、RankBM25、Sentence-BERT、CoSENT等文本表征、文本相似度计算模型,开箱即用。
https://pypi.org/project/text2vec/
Apache License 2.0
4.48k stars 396 forks source link

添加Bfloat16和多卡DataParallel训练功能 #100

Closed wptoux closed 1 year ago

wptoux commented 1 year ago

添加Bfloat16和多卡DataParallel训练功能,用于训练更大的向量模型。

Add Bfloat16 and DataParallel training capabilities for training larger vector models.

shibing624 commented 1 year ago

done