适用于昇腾 NPU 的 mobilenetv3_small_075 模型卡片。
import timm
import torch
model = timm.create_model("mobilenetv3_small_075", pretrained=True, num_classes=1000)
model = model.to("npu:0")
model.eval()
# Run inference
with torch.no_grad():
output = model(input_tensor)经验证,该模型在 CPU 和 NPU 后端上的输出匹配度误差在 1% 以内。
Apache 2.0