xujiashuai/convnext_nano
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

convnext_nano.d1h_in1k - Ascend NPU 推理

模型信息

  • 模型: convnext_nano.d1h_in1k
  • 框架: PyTorch + torch_npu
  • 硬件: Ascend 910B4 NPU
  • 任务: 图像分类

推理输出证据

/usr/local/python3.11.14/lib/python3.11/site-packages/torch_npu/utils/collect_env.py:58: UserWarning: Warning: The /usr/local/Ascend/cann-8.5.1 owner does not match the current owner.
  warnings.warn(f"Warning: The {path} owner does not match the current owner.")
/usr/local/python3.11.14/lib/python3.11/site-packages/torch_npu/utils/collect_env.py:58: UserWarning: Warning: The /usr/local/Ascend/cann-8.5.1/aarch64-linux/ascend_toolkit_install.info owner does not match the current owner.
  warnings.warn(f"Warning: The {path} owner does not match the current owner.")
/usr/local/python3.11.14/lib/python3.11/site-packages/torch_npu/utils/_path_manager.py:66: UserWarning: Permission mismatch: The owner of /usr/local/python3.11.14/lib/python3.11/site-packages/torch_npu/lib/libop_plugin_atb.so does not match.
  warnings.warn(f"Permission mismatch: The owner of {path} does not match.")
/usr/local/python3.11.14/lib/python3.11/site-packages/torch_npu/utils/collect_env.py:58: UserWarning: Warning: The /usr/local/Ascend/cann-8.5.1 owner does not match the current owner.
  warnings.warn(f"Warning: The {path} owner does not match the current owner.")
/usr/local/python3.11.14/lib/python3.11/site-packages/torch_npu/utils/collect_env.py:58: UserWarning: Warning: The /usr/local/Ascend/cann-8.5.1/aarch64-linux/ascend_toolkit_install.info owner does not match the current owner.
  warnings.warn(f"Warning: The {path} owner does not match the current owner.")
模型: convnext_nano.d1h_in1k
设备: npu:0
精度: float32
------------------------------------------------------------
权重加载完成

--- CPU 推理 ---
输出形状: [1, 1000]
Logits 前5个: [-0.84898525 -0.04340765 -0.05851972 -0.49776074  0.18654487]
是否有 NaN: False
预测类别: 644

--- NPU 推理 (npu:0) ---
输出形状: [1, 1000]
Logits 前5个: [-0.8470695  -0.02410723 -0.0448494  -0.48037645  0.19392367]
是否有 NaN: False
预测类别: 644

--- 精度对比 ---
Cosine Similarity: 0.999684
Max Abs Error: 0.040346

--- 性能基准 ---
平均延迟: 5.58 ms (10轮)

✓ 推理完成
[LOG_WARNING] can not create directory, directory: /home/atomgit/ascend/log, possible reason: No such file or directory.path string is NULLpath string is NULL

使用方法

python inference.py --device npu:0 --dtype float32