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

convnext_pico.d1_in1k - Ascend NPU 推理

模型信息

  • 模型: convnext_pico.d1_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_pico.d1_in1k
设备: npu:0
精度: float32
------------------------------------------------------------
权重加载完成

--- CPU 推理 ---
输出形状: [1, 1000]
Logits 前5个: [-1.2267065   0.12814981  0.32091102 -0.33794367 -0.03496172]
是否有 NaN: False
预测类别: 644

--- NPU 推理 (npu:0) ---
输出形状: [1, 1000]
Logits 前5个: [-1.2484086   0.13994765  0.34723014 -0.32453632 -0.02763061]
是否有 NaN: False
预测类别: 644

--- 精度对比 ---
Cosine Similarity: 0.999738
Max Abs Error: 0.068066

--- 性能基准 ---
平均延迟: 4.50 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