xujiashuai/maxvit_rmlp_nano_rw_256_sw
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

maxvit_rmlp_nano_rw_256_sw - Ascend NPU 适配

1. 模型信息

  • 模型: maxvit_rmlp_nano_rw_256_sw
  • 任务: 图像分类 / 模型适配

2. 昇腾 NPU 适配结果

指标值
设备Ascend NPU (npu:0)
精度float32
Cosine Similarity0.999994
Max Absolute Error0.006382
NPU 推理时间15.45ms

3. 推理脚本

python3 inference.py --device npu:0

4. 推理输出证据

============================================================
  模型: maxvit_rmlp_nano_rw_256_sw
  时间: 2026-05-19 11:00:01
  设备: Ascend NPU (npu:0)
============================================================

=== maxvit_rmlp_nano_rw_256.sw_in1k ===
Dtype: torch.float32

--- CPU 推理 ---
输出形状: torch.Size([1, 1000])
输出前5值: [-0.1770019829273224, -0.07608211040496826, 0.489001989364624, 0.2834790349006653, 0.32106906175613403]
推理时间: 293.61ms

--- NPU 推理 ---
输出形状: torch.Size([1, 1000])
输出前5值: [-0.1739749163389206, -0.0754166841506958, 0.48960167169570923, 0.2846854627132416, 0.32059144973754883]
推理时间: 15.19ms

=== 精度对比 ===
Cosine Similarity: 0.999991
Max Absolute Error: 0.005028
Relative Error: 0.148568

✓ NPU 适配通过

5. 精度校验

CPU 与 NPU 推理结果对比：

  • Cosine Similarity: 0.999994 (> 0.99 通过)
  • Max Absolute Error: 0.006382
  • 状态: SUCCESS

6. 文件说明

  • inference.py - NPU 推理脚本
  • report.json - 精度验证报告
  • screenshots/ - 推理输出截图