CoROM 模型用于中文段落排序,基于医疗领域数据训练。
iic/nlp_corom_passage-ranking_chinese-tiny-medical| 项目 | 数值 |
|---|---|
| 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 |
Result: {'scores': [0.994, 0.015, 0.957]}
SUCCESSpython3 inference.py
python3 evaluate.pyinference.py - NPU 推理脚本evaluate.py - CPU 与 NPU 余弦相似度评估