cait_xxs24_224_fb_dist on Ascend NPU
1. 简介
- 模型来源: cait_xxs24_224_fb_dist
- 适配状态: SUCCESS
- 适配时间: 2026-05-19 11:11:19
2. 验证环境
| 组件 | 版本 |
|---|
| torch | 2.9.0 |
| torch-npu | 2.9.0.post1 |
| CANN | 8.5.1 |
| NPU | Ascend 910B4 |
| OS | Ubuntu 22.04.5 LTS (aarch64) |
| Python | 3.11.14 |
3. 精度评测
✅ NPU vs CPU 精度对比
| 指标 | 数值 |
|---|
| Cosine Similarity | 0.999993 |
| Max Absolute Error | 0.012272 |
| 精度误差 | 0.01% |
| 是否满足要求 | 是（< 1%）✅ |
4. 性能
5. 推理输出证据
============================================================
模型: cait_xxs24_224_fb_dist
时间: 2026-05-19 10:46:48
设备: Ascend NPU (npu:0)
============================================================
=== cait_xxs24_224.fb_dist_in1k ===
Dtype: torch.float32
--- CPU 推理 ---
输出形状: torch.Size([1, 1000])
输出前5值: [0.5715703964233398, -0.2678225338459015, -0.8648203611373901, 0.48628371953964233, 1.1819676160812378]
推理时间: 306.30ms
--- NPU 推理 ---
输出形状: torch.Size([1, 1000])
输出前5值: [0.5715355277061462, -0.26772308349609375, -0.8636564016342163, 0.4834529459476471, 1.1811643838882446]
推理时间: 14.88ms
=== 精度对比 ===
Cosine Similarity: 0.999992
Max Absolute Error: 0.011150
Relative Error: 0.064934
✓ NPU 适配通过
6. 复现步骤
# 环境准备
source /usr/local/Ascend/ascend-toolkit/set_env.sh
# 运行推理
python3 inference.py --device npu:0
7. 评测材料
| 材料 | 文件 |
|---|
| 推理脚本 | inference.py |
| 精度报告 | report.json |
| 截图证据 | screenshots/ |
贡献者: xujiashuai | 赛道: 模型适配赛道 | 时间: 2026-05-19 11:11:19
