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

timm/deit_base_patch16_224.fb_in1k - 昇腾NPU适配

1. 模型概述

DeiT-Base Patch16 224（Facebook）视觉Transformer

  • 原始模型：timm/deit_base_patch16_224.fb_in1k
  • 框架：PyTorch（timm）
  • 任务：图像分类（ImageNet-1K）
  • 参数：86567656
  • 状态：成功

2. 环境

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

3. 精度

指标值
余弦相似度0.999886
最大绝对误差0.011629
CPU/NPU预测结果匹配（类别138）

4. 性能

指标值
平均延迟5.16 ms
输出形状[1, 1000]

5. CPU与NPU对比

指标CPUNPU差异
余弦相似度1.00.999886-
输出形状[1,1000][1,1000]相同
预测类别138138相同