- 博客(1)
- 收藏
- 关注
RSS订阅原创 在C++平台部署pytorch模型的两种方法
1、准备工作 1、模型转换代码： import torch from model.FCN_Aux import FCN_AUX model = FCN_AUX(2) model_weight_path = './model/better.pth'#训练完模型位置 model.load_state_dict(torch.load(model_weight_path)) model.cuda() model.eval() example = torch.rand(1, 3, 224, 224).cuda() t
2020-05-22 22:24:18
3240
7
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人