CoROM 是一款基于 BERT 的轻量级句向量模型,专为电商领域优化。
iic/nlp_corom_sentence-embedding_chinese-tiny-ecom| 项目 | 数值 |
|---|---|
| NPU | Ascend910B4 |
| CANN | 8.5.1 |
| PyTorch | 2.9.0+cpu |
| torch_npu | 2.9.0.post1 |
| Python | 3.11.14 |
CPU 与 NPU 余弦相似度:
| 指标 | 数值 |
|---|---|
| 余弦相似度 | 1.000000 |
Embedding shape: [1, 256]
Embedding sample: [0.1568, -1.2536, 0.1713, -0.5272, -0.1829]
SUCCESSpython3 inference.py
python3 evaluate.pyinference.py - NPU 推理脚本evaluate.py - CPU 与 NPU 余弦相似度评估