PyTorch模型加载与保存的最佳实践
摘要：
一般来说PyTorch有两种保存和读取模型参数的方法。但这篇文章我记录了一种最佳实践，可以在加载模型时避免掉一些问题。 传统方案： 第一种方案是保存整个模型： torch.save(model_object, 'model.pth') 第二种方法是保存模型网络参数： torch.save(model 阅读全文
posted @ 2020-05-18 19:54 那抹阳光1994 阅读(1068) 评论(0) 推荐(0) 编辑
