ai-for-science-deepfri-tf-npu
Installation
SKILL.md
DeepFRI TensorFlow 原生昇腾 NPU 迁移 Skill
项目概述
DeepFRI 是 Flatiron Institute 开发的蛋白质功能预测框架，原始实现基于 TensorFlow/Keras。本 Skill 记录使用 TF Community 方式（TF 2.6.5 + npu_device） 将其直接部署到昇腾 NPU 上，无需转换为 PyTorch。
与 TF→PyTorch 迁移方式的区别：
| 维度 | TF→PyTorch 迁移 | TF Community 迁移（本 Skill） |
|---|---|---|
| 代码改动量 | 大（重写模型+转换权重） | 极小（仅添加 2 行初始化） |
| 依赖框架 | PyTorch + torch_npu | TensorFlow 2.6.5 + npu_device |
| 编译需求 | 无 | 需源码编译 TF（aarch64） |
| GCN 模型支持 | 需额外处理 CuDNNLSTM | 直接支持（CPU 版模型） |
| 精度 | < 1e-4（FP32 完全匹配） | < 1e-2（HF32 混合精度） |