xujiashuai/timm-mvitv2_small.fb_in1k
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

timm-mvitv2_small.fb_in1k 在昇腾 NPU 上的适配

1. 简介

  • 模型来源:timm/mvitv2_small.fb_in1k
  • 架构:MViTv2-Small(多尺度视觉Transformer)
  • 参数量:34,870,216
  • 任务:图像分类(ImageNet-1k)
  • 适配状态:成功(SUCCESS)
  • 适配时间:2026-05-17

2. 验证环境

组件版本
torch2.9.0
torch-npu2.9.0.post1
CANN8.5.1
NPUAscend 910B4

3. 精度评测

指标数值
余弦相似度0.999965
最大绝对误差0.004244
是否满足要求是(< 1%)

4. 性能

指标数值
平均延迟21.49 毫秒
输出形状[1, 1000]

5. 适配说明

使用 timm 库 timm.create_model('mvitv2_small.fb_in1k', pretrained=True) 加载预训练权重。输入为随机图像张量 (1, 3, 224, 224),比较分类 logits 输出。

6. 推理输出证据

Model: timm-mvitv2_small.fb_in1k
Output shape: [1, 1000]
Latency: 21.49ms
Cosine Similarity: 0.999965
Max Abs Error: 0.004244

7. CPU vs NPU 精度对比

指标CPUNPU差异
余弦相似度1.00.9999650.0035%
最大绝对误差-0.004244-