本仓库包含了 timm 中的 vit_base_patch32_clip_448.laion2b_ft_in12k_in1k 模型在华为昇腾 NPU 上的适配与推理实现。
vit_base_patch32_clip_448.laion2b_ft_in12k_in1k (timm / HuggingFace Hub)本模型基于 timm 库加载预训练权重,模型权重通过 ModelScope 下载。在昇腾 NPU 上通过 torch_npu 实现硬件加速。
pip install -r requirements.txtpython inference.pypip install -r requirements.txtpython inference.pypython compare_cpu_npu.py| 指标 | 数值 |
|---|---|
| CPU 推理耗时 | 0.6903s |
| NPU 推理耗时 | 0.0056s |
| MAE (平均绝对误差) | 0.00476959 |
| Max Error (最大误差) | 0.02575183 |
| MSE (均方误差) | 3.697e-05 |
| 相对误差 | 0.00448046 |
| 误差百分比 | 0.448 |
| 余弦相似度 | 0.99998998 |
| Top-5 索引一致性 | 是 |
结论: NPU 与 CPU 推理误差 < 1%
| 设备 | 推理耗时 (s) |
|---|---|
| CPU | 0.6903 |
| NPU | 0.0056 |
CPU inference time: 0.6903s
Top-5 indices: [551, 838, 921, 892, 409]
Top-5 values: [6.384991, 5.199828, 5.108638, 4.745762, 4.658279]
NPU inference time: 0.0056s
Top-5 indices: [551, 838, 921, 892, 409]
Top-5 values: [6.371974, 5.200135, 5.113847, 4.746722, 4.659467]
CPU/NPU Relative Error: 0.448%Mean Absolute Error (MAE): 0.00476959
Max Absolute Error: 0.02575183
Cosine Similarity: 0.99998998
Relative Error: 0.00448046运行截图由 acend- 工具生成,展示推理成功画面。