2020年6月15日
[pytorch]模型参数保存与加载
摘要： 最简单的情况 模型保存： torch.save(model.state_dict(), PATH) 模型加载： model.load_state_dict(torch.load(PATH)) 此时保存的是一个字典，key为model中的weight或bias名，如"linear1.weight"或“ 阅读全文
posted @ 2020-06-15 08:40 DinoX 阅读(639) 评论(0) 推荐(0) 编辑