pytorch android
【摘要】
导出的模型Android能加载:
速度是pc的5分之一
注意:
1.需要pytorch 1.3以上版本,
2.eval不能少,少了check会报错
import torchimport torchvision model = torchvision.models.resnet18(pretrained=True)model.eval()example...
导出的模型Android能加载:
速度是pc的5分之一
注意:
1.需要pytorch 1.3以上版本,
2.eval不能少,少了check会报错
-
import torch
-
import torchvision
-
-
model = torchvision.models.resnet18(pretrained=True)
-
model.eval()
-
example = torch.rand(1, 3, 224, 224)
-
traced_script_module = torch.jit.trace(model, example)
-
traced_script_module.save("app/src/main/assets/model.pt")
demo参考:
https://pytorch.org/mobile/android/
文章来源: blog.csdn.net,作者:网奇,版权归原作者所有,如需转载,请联系作者。
原文链接:blog.csdn.net/jacke121/article/details/102731987
【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)