swin_tiny_patch4_window7_224 NPU 模型

适用于 Ascend NPU 的 swin_tiny_patch4_window7_224 模型卡片。

模型说明

架构：Swin Transformer Tiny
任务：图像分类
框架：timm
硬件：Ascend NPU

使用方法

import timm
import torch

model = timm.create_model("swin_tiny_patch4_window7_224", pretrained=True, num_classes=1000)
model = model.to("npu:0")
model.eval()

# Run inference
with torch.no_grad():
    output = model(input_tensor)

性能

经验证，该模型在CPU和NPU后端上的输出匹配度在1%容差范围内。

许可证

Apache 2.0