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

deit_tiny_distilled_patch16_224 - Ascend NPU 推理

模型信息

  • 模型: deit_tiny_distilled_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_distilled_patch16_224.fb_in1k/resolve/main/model.safetensors (Caused by ConnectTimeoutError(<HTTPSConnection(host='hf-mirror.com', port=443) at 0xfffe7e62abd0>, 'Connection to hf-mirror.com timed out. (connect timeout=10)'))"), '(Request ID: 85ee7284-280d-4f5e-b2c5-8812285a3cdf)')' thrown while requesting HEAD https://hf-mirror.com/timm/deit_tiny_distilled_patch16_224.fb_in1k/resolve/main/model.safetensors
Retrying in 1s [Retry 1/5].
模型: deit_tiny_distilled_patch16_224
设备: npu:0
精度: float32
------------------------------------------------------------
权重加载完成

--- CPU 推理 ---
输出形状: [1, 1000]
Logits 前5个: [ 0.6244517  -0.77603173  0.73568153  0.12734553  1.309432  ]
是否有 NaN: False
预测类别: 539

--- NPU 推理 (npu:0) ---
输出形状: [1, 1000]
Logits 前5个: [ 0.63507366 -0.7709336   0.7257421   0.12058454  1.31851   ]
是否有 NaN: False
预测类别: 539

--- 精度对比 ---
Cosine Similarity: 0.999974
Max Abs Error: 0.024033

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