ai-for-science-proteinbert
Installation
SKILL.md
ProteinBERT 昇腾 NPU 部署 Skill
将 ProteinBERT 从 TensorFlow/Keras 完整迁移到 PyTorch + torch_npu， 已通过 5 个基准任务的精度验证（以 GPU 为基线）。
模型概况
| 项目 | 值 |
|---|---|
| 架构 | 6 层 Conv + GlobalAttention blocks |
| 参数量 | ~16M (vocab=26, annotations=8943, d_seq=128, d_global=512) |
| 原始框架 | TensorFlow 2.x / Keras |
| 目标框架 | PyTorch 2.5.1 + torch_npu 2.5.1 |
| 验证 CANN | 8.3.RC1 |
| 硬件 | Ascend910 |