摘要： 一般来说PyTorch有两种保存和读取模型参数的方法。但这篇文章我记录了一种最佳实践，可以在加载模型时避免掉一些问题。 第一种方案是保存整个模型： 1 torch.save(model_object, 'model.pth') 第二种方法是保存模型网络参数： 1 torch.save(model_o 阅读全文