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

deit_tiny_patch16_224 - Ascend NPU 推理

模型信息

  • 模型: deit_tiny_patch16_224
  • 框架: 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.")
'(MaxRetryError("HTTPSConnectionPool(host='hf-mirror.com', port=443): Max retries exceeded with url: /timm/deit_tiny_patch16_224.fb_in1k/resolve/main/model.safetensors (Caused by ConnectTimeoutError(<HTTPSConnection(host='hf-mirror.com', port=443) at 0xfffe7793f390>, 'Connection to hf-mirror.com timed out. (connect timeout=10)'))"), '(Request ID: 7a604309-616d-4b21-bc2e-69da2da6078d)')' thrown while requesting HEAD https://hf-mirror.com/timm/deit_tiny_patch16_224.fb_in1k/resolve/main/model.safetensors
Retrying in 1s [Retry 1/5].
模型: deit_tiny_patch16_224
设备: npu:0
精度: float32
------------------------------------------------------------
权重加载完成

--- CPU 推理 ---
输出形状: [1, 1000]
Logits 前5个: [-0.14378785  0.64892715  0.16337629  0.33603436  0.71451676]
是否有 NaN: False
预测类别: 963

--- NPU 推理 (npu:0) ---
输出形状: [1, 1000]
Logits 前5个: [-0.1441636   0.6453378   0.15925407  0.33213684  0.70980656]
是否有 NaN: False
预测类别: 963

--- 精度对比 ---
Cosine Similarity: 0.999994
Max Abs Error: 0.007913

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