- 博客(1)
- 收藏
- 关注
原创 pytorch训练的pt模型转换为onnx(nn.DataParallel()、model、model.state_dict())
1、读取pt模型文件,文件既可以是torch.save(model,path)整体保存的模型,也可以是保存的字典文件。2、torch.onnx.export()函数转换。
2022-08-29 16:28:49 1983 2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
1、读取pt模型文件,文件既可以是torch.save(model,path)整体保存的模型,也可以是保存的字典文件。2、torch.onnx.export()函数转换。
2022-08-29 16:28:49 1983 2
TA创建的收藏夹 TA关注的收藏夹
TA关注的人