CoROM 段落排序（nlp_corom_passage-ranking_chinese-tiny）

用于中文段落排序的 CoROM 模型，基于通用领域数据训练。

模型 ID：iic/nlp_corom_passage-ranking_chinese-tiny
任务：文本排序
框架：PyTorch
隐藏层大小：256
层数：4

验证环境

项目	数值
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]}
SUCCESS

复现

python3 inference.py
python3 evaluate.py

材料

inference.py - NPU 推理脚本
evaluate.py - CPU 与 NPU 余弦相似度评估