- 博客(1)
- 收藏
- 关注
RSS订阅原创 在C++平台部署pytorch模型的两种方法
1、准备工作1、模型转换代码：import torchfrom model.FCN_Aux import FCN_AUXmodel = 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
3121
7
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人