search

shibing624 / text2vec

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

模型微调卡住 #139

Closed CathyKitten closed 10 months ago

CathyKitten commented 10 months ago

想请问一下，用python training_sup_text_matching_model_jsonl_data.py --model_arch cosent --do_train --do_predict --num_epochs 10 --model_name hfl/chinese-macbert-base --output_dir ./outputs/STS-B-cosent命令跑的时候，会停在下面这个界面一段时间吗。跑的速度会比较慢吗。 image

shibing624 commented 10 months ago

不是bug，cpu训练bert类模型极慢。