timm/swin_small_patch4_window7_224.ms_in22k - 昇腾NPU

Swin Transformer Small在ImageNet-22k（21841个类别）上进行预训练，通过PyTorch + torch_npu适配单卡华为昇腾NPU推理。

项目	数值
模型	timm/swin_small_patch4_window7_224.ms_in22k
架构	Swin Transformer Small
预训练	ImageNet-22k（微软）
输入分辨率	224x224
num_classes	21841
硬件	华为昇腾NPU

推理

pip install -r requirements.txt
python3 inference.py

assets/test.jpg的Top-5预测结果：

python3 eval_accuracy.py

单张测试图像上的CPU与NPU一致性检查：

python3 benchmark.py

昇腾NPU上的单图延迟（运行10次，批次大小=1）：

Self-Verification

#NPU

Swin Transformer Small在ImageNet-22k（21841个类别）上进行预训练，通过PyTorch + torch_npu适配单卡华为昇腾NPU推理。

项目	数值
模型	timm/swin_small_patch4_window7_224.ms_in22k
架构	Swin Transformer Small
预训练	ImageNet-22k（微软）
输入分辨率	224x224
num_classes	21841
硬件	华为昇腾NPU

pip install -r requirements.txt
python3 inference.py

assets/test.jpg的Top-5预测结果：

python3 eval_accuracy.py

单张测试图像上的CPU与NPU一致性检查：

python3 benchmark.py

昇腾NPU上的单图延迟（运行10次，批次大小=1）：

Self-Verification

#NPU