BERT对话检测模型（speech_bert_dialogue-detetction_speaker-diarization_chinese）

基于BERT的说话人 diarization 场景下的对话检测模型。

• 模型ID：iic/speech_bert_dialogue-detetction_speaker-diarization_chinese
• 任务：说话人 diarization - 对话检测
• 框架：PyTorch
• 输入：中文文本（转录语音）
• 输出：二分类 logits（2 类）

验证环境

精度评估

CPU 与 NPU 余弦相似度：

推理输出

Logits shape: [1, 2]
Logits sample: [2.089, -2.814]
SUCCESS

复现

python3 inference.py
python3 evaluate.py

材料

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