MobileOne首頁、文檔和下載- 移動端毫秒Backbone 改進- 程式开发

MobileOne 是蘋果公司改進的一毫秒移動骨幹網(Backbone)模型,該模型在1 毫秒內實現了75.9% 的Top-1 精度。

Model Zoo

在iPhone 12 Pro 上測得的延遲

用法

要使用模型,請遵循以下代碼片段,

import torch
from mobileone import mobileone, reparameterize_model

# To Train from scratch/fine-tuning
model = mobileone(variant='s0')
# ... train ...

# Load Pre-trained checkpoint for fine-tuning
checkpoint = torch.load('/path/to/unfused_checkpoint.pth.tar')
model.load_state_dict(checkpoint)
# ... train ...

# For inference
model.eval()      
model_eval = reparameterize_model(model)
# Use model_eval at test-time

要簡單地評估模型,請使用分支重新參數化的融合檢查點。

import torch
from mobileone import mobileone

model = mobileone(variant='s0', inference_mode=True)
checkpoint = torch.load('/path/to/checkpoint.pth.tar')
model.load_state_dict(checkpoint)
# ... evaluate/demo ...

#MobileOne首頁文檔和下載 #移動端毫秒Backbone #改進 #程式开发

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *