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

timm/swinv2_tiny_window8_256.ms_in1k - 昇腾NPU适配

1. 模型概述

SwinV2 Tiny是一种采用移位窗口的分层视觉Transformer，窗口大小为8，在ImageNet-1k上以256x256分辨率进行训练。

  • 原始模型：timm/swinv2_tiny_window8_256.ms_in1k
  • 架构：SwinV2 Transformer
  • 参数数量：28,347,154
  • 输入分辨率：256x256
  • 输出类别：1000

2. NPU适配

  • 设备：昇腾NPU（npu:0）
  • 框架：PyTorch + torch_npu

3. 精度

指标数值
余弦相似度0.999986
最大绝对误差0.002450

4. 延迟

设备平均延迟
NPU（npu:0）12.51 ms

5. CPU与NPU对比

指标CPUNPU
输出形状[1, 1000][1, 1000]
预测结果852852

6. 使用方法

source setup_env.sh
HF_ENDPOINT=https://hf-mirror.com python3 inference.py --device npu:0