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

timm/swinv2_cr_small_ns_224.sw_in1k - 昇腾NPU适配

1. 模型概述

SwinV2-CR Small是一种紧凑重组的Swin Transformer V2模型，它在ImageNet-1k数据集上以224x224分辨率结合邻域注意力进行训练。

  • 原始模型：timm/swinv2_cr_small_ns_224.sw_in1k
  • 架构：SwinV2-CR（紧凑重组）
  • 参数数量：49,696,444
  • 输入分辨率：224x224
  • 输出类别：1000

2. NPU适配

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

3. 精度

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

4. 延迟

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

5. CPU与NPU对比

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

6. 使用方法

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