MobileOne移动端毫秒 Backbone 改进

联合创作 · 2023-09-26 06:58

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 ...
浏览 3
点赞
评论
收藏
分享

手机扫一扫分享

编辑
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑
举报