基于BERT的说话人分轨场景下对话检测模型。
iic/speech_bert_dialogue-detetction_speaker-diarization_chinese| 项目 | 数值 |
|---|---|
| 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 |
Logits shape: [1, 2]
Logits sample: [2.089, -2.814]
SUCCESSpython3 inference.py
python3 evaluate.pyinference.py - NPU 推理脚本evaluate.py - CPU 与 NPU 余弦相似度评估