wangyuxinwhy / uniem

unified embedding model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

复现M3E-Base

hjq133 opened this issue · comments

commented

如果想要复现M3E-Base的结果,不知道作者方便透露下具体的参数或者相应的config file吗,
还是说train_m3e.py里的默认参数已经是最优参数了?
感谢!

temperature=0.01
batch_size=80
epochs=1
mixed_precision=fp16
其他都是默认的参数

commented

感谢!

请问训练数据集方便share吗?

请问训练数据集方便share吗?

https://huggingface.co/moka-ai/m3e-base#%F0%9F%93%82-m3e%E6%95%B0%E6%8D%AE%E9%9B%86

可以查看这个链接