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

timm/regnety_320.tv2_in1k - 昇腾NPU适配

1. 模型概述

RegNetY-320是源自TorchVision v2方案的大型RegNet架构模型，在ImageNet-1k上训练，包含挤压激励模块。

  • 原始模型：timm/regnety_320.tv2_in1k
  • 架构：RegNetY（含SE模块）
  • 参数数量：145,046,770
  • 输入分辨率：224x224
  • 输出类别：1000（ImageNet-1k）

2. NPU适配

  • 设备：昇腾NPU（npu:0）
  • 框架：PyTorch + torch_npu
  • 推理脚本：inference.py

3. 精度

指标数值
余弦相似度1.000000
最大绝对误差0.002933

4. 延迟

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

5. CPU与NPU对比

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

6. 使用方法

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