facebook/regnet-x-040 on Ascend NPU
1. 简介
本文档记录了 facebook/regnet-x-040 模型在华为昇腾 NPU (Ascend910) 上的适配过程和验证结果。
RegNet-X 是 Meta 提出的基于规则的网络架构，通过设计空间搜索得到的高效卷积神经网络。X 变体使用标准卷积操作。
2. 验证环境
- 硬件: 华为昇腾 910 NPU
- 软件: CANN 8.5.1, PyTorch 2.x, torch_npu
- 操作系统: Linux
- 模型来源: ModelScope (facebook/regnet-x-040)
- 权重格式: pytorch_model.bin
3. 推理运行
pip install -r requirements.txt
python inference.py
推理结果 (NPU Top-5):
- Top-1: bathing cap, swimming cap (0.7381)
- Top-2: sea lion (0.0620)
- Top-3: grey whale, gray whale, devilfish, Eschrichtius gibbosus, Eschrichtius robustus (0.0578)
- Top-4: maillot (0.0333)
- Top-5: swimming trunks, bathing trunks (0.0207)
日志保存在 logs/inference.log。
4. 精度验证
对单张测试图片进行 CPU 与 NPU 一致性验证：
| 指标 | 数值 |
|---|
| max_abs_error | 0.004403 |
| mean_abs_error | 0.001036 |
| relative_error | 0.0521% |
| cosine_similarity | 1.000000 |
| threshold | 1.0% |
| 结果 | PASS |
- CPU Top-1: bathing cap, swimming cap
- NPU Top-1: bathing cap, swimming cap
- CPU Top-5: bathing cap, swimming cap, sea lion, grey whale, maillot, swimming trunks
- NPU Top-5: bathing cap, swimming cap, sea lion, grey whale, maillot, swimming trunks
- Top-1 match: True
- Top-5 match: True
5. 性能参考
| 指标 | 数值 |
|---|
| avg_latency | 7.65 ms |
| min_latency | 7.51 ms |
| max_latency | 7.75 ms |
| throughput | 130.66 images/sec |
6. 精度评测说明
本项目包含单图 smoke consistency 验证，非官方 ImageNet 完整验证集评测。详细指标见第 4 节。
7. 自验证截图
详见 screenshots/self_verification.png 和 screenshots/self_verification.txt。
8. 日志文件
logs/env_check.log - NPU 环境检查logs/inference.log - 推理结果logs/accuracy.log - 精度验证logs/benchmark.log - 性能测试
9. 注意事项
- 本适配使用 ModelScope snapshot_download 下载权重，避免直接从 HuggingFace 下载
- 模型使用 transformers AutoModelForImageClassification 加载
- 精度验证采用单图 smoke test，非完整 ImageNet 验证集评测
- NPU 推理结果与 CPU 结果高度一致，相对误差 < 1.0%
10. 标签
#NPU #Ascend #Ascend910 #RegNet #ImageClassification #facebook