2019年5月24日
Pytorch 模型的加载与保存
摘要： <! flowchart 箭头图标 勿删 pytorch的模型和参数是分开的，可以分别保存或加载模型和参数。 1、直接保存模型 保存模型 torch.save(model, 'model.pth') 加载模型 model = torch.load('model.pth') 2、分别加载模型的结构和参 阅读全文
posted @ 2019-05-24 20:50 Le1B_o 阅读(12644) 评论(1) 推荐(0)