XiaYuanOwO/ascend-deit-tiny-patch16-224-model
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

deit-tiny-patch16-224

基本信息

项目内容
模型名称deit-tiny-patch16-224
原始模型timm/deit_tiny_patch16_224
任务类型image classification
硬件环境Ascend 910 NPU
框架PyTorch 2.x + torch_npu + timm

适配说明

本项目将 timm/deit_tiny_patch16_224 模型适配至昇腾 NPU 平台,通过 torch_npu 实现 CPU/NPU 精度对齐与推理验证。

环境依赖

  • Python 3.8+
  • torch >= 2.0.0
  • torch_npu >= 2.0.0
  • timm
  • Pillow
  • numpy

权重说明

模型权重通过 timm 官方渠道(huggingface hub)下载,不提交至本仓库。 首次运行 inference.py 时自动下载缓存。

不提交权重文件

本仓库仅包含推理代码与适配脚本,不提交任何模型权重文件(.bin, .safetensors, .pt, .ckpt)。

NPU 推理

python inference.py

CPU/NPU 精度对比

指标值
Cosine Similarity1.0
Mean Abs Diff3.4e-05
MatchTrue

性能数据

环境延迟
CPU64.92ms
NPU189.26ms

适配结论

模型在昇腾 NPU 上推理结果与 CPU 一致,精度满足要求。

#+NPU