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 ...
评论